IrwGAN
IrwGAN copied to clipboard
testing error
Hello, I am a deep learning newbie currently testing your model on a dataset as a proposal for an MS project but when I run it using the provided code, I get this: /usr/local/lib/python3.7/dist-packages/torchvision/transforms/transforms.py:333: UserWarning: Argument interpolation should be of type InterpolationMode instead of int. Please, use InterpolationMode enum. "Argument interpolation should be of type InterpolationMode instead of int. " model [IRWGANModel] was created
Information
task: dibcoirwgan
phase: test
gan_type: lsgan
netD: gl
trainA_size: 93
trainB_size: 84
testA_size: 16
testB_size: 16
Weight
lambda_A: 10
lambda_B: 10
lambda_identity: 1
Model Specific
beta_mode: AB
threshold: 0.1
batch_size: 20
lambda_nos_A: 1
lambda_nos_B: 1
-------------- Networks loaded ---------------- [Network gen_a2b] Total number of parameters : 11.379 M [Network gen_b2a] Total number of parameters : 11.379 M [Network dis_a] Total number of parameters : 13.969 M [Network dis_b] Total number of parameters : 13.969 M [Network beta_net_a] Total number of parameters : 2.757 M [Network beta_net_b] Total number of parameters : 2.757 M
[*] testing start!
Traceback (most recent call last):
File "main.py", line 117, in
has anyone ever come across this?
Hi, it looks like that some directories are empty. Could you please go to models/misc.py 125 and add print(fake_b_path, real_b_path) and print(fake_a_path, real_a_path) ?
If you are sure that you have images in these four directories, I will have a look this weekend.