[torchbench] pytorch_CycleGAN_and_pix2pix fails to run.

[ysiraichi]

🐛 Bug

Running the upstreamed benchmarking scripts with the following command results in an unexpected error.

python xla/benchmarks/experiment_runner.py \
       --suite-name torchbench \
       --accelerator cuda \
       --xla PJRT --xla None \
       --dynamo openxla --dynamo None \
       --test eval --test train \
       --repeat 30 --iterations-per-run 5 \
       --print-subprocess \
       --no-resume -k pytorch_CycleGAN_and_pix2pix

usage: experiment_runner.py [-h] --dataroot DATAROOT [--name NAME]
                            [--gpu_ids GPU_IDS]
                            [--checkpoints_dir CHECKPOINTS_DIR]
                            [--model MODEL] [--input_nc INPUT_NC]
                            [--output_nc OUTPUT_NC] [--ngf NGF] [--ndf NDF]
                            [--netD NETD] [--netG NETG]
                            [--n_layers_D N_LAYERS_D] [--norm NORM]
                            [--init_type INIT_TYPE] [--init_gain INIT_GAIN]
                            [--no_dropout] [--dataset_mode DATASET_MODE]
                            [--direction DIRECTION] [--serial_batches]
                            [--num_threads NUM_THREADS]
                            [--batch_size BATCH_SIZE] [--load_size LOAD_SIZE]
                            [--crop_size CROP_SIZE]
                            [--max_dataset_size MAX_DATASET_SIZE]
                            [--preprocess PREPROCESS] [--no_flip]
                            [--display_winsize DISPLAY_WINSIZE]
                            [--epoch EPOCH] [--load_iter LOAD_ITER]
                            [--verbose] [--suffix SUFFIX]
                            [--results_dir RESULTS_DIR]
                            [--aspect_ratio ASPECT_RATIO] [--phase PHASE]
                            [--eval] [--num_test NUM_TEST]
                            [--model_suffix MODEL_SUFFIX]
experiment_runner.py: error: unrecognized arguments:

Environment

• PyTorch/XLA: 402166b

Additional context

This might be related to how this model is initialized.