I train module with my own datasets and write a bash like below
"
python main.py --mode test --dataset RaFD --image_size 240 --c_dim 5 --rafd_image_dir AMOS_Data/00004534 --sample_dir stargan_custom/samples--log_dir stargan_custom/logs --model_save_dir stargan_custom/models --result_dir stargan_custom/results
"
However, here is the ERROR after running:
"from torchvision import transforms as T
...
transform.append(T.Resize(image_size))
..."
AttributeError: 'module' object has no attribute 'Resize'
Could anyone please?