Hello,
I am trying to train on the jean2skirt dataset and after the following train command:
python train.py --dataroot ./datasets/jeans2skirt_ccp --model insta_gan --name jeans2skirt_ccp_instagan --loadSizeH 330 --loadSizeW 220 --fineSizeH 30 --fineSizeW 20 --gpu_ids -1
I get the following error:
----------------- Options ---------------
batch_size: 1
beta1: 0.5
checkpoints_dir: ./checkpoints
continue_train: False
dataroot: ./datasets/jeans2skirt_ccp [default: None]
dataset_mode: unaligned_seg
direction: AtoB
display_env: main
display_freq: 400
display_id: 1
display_ncols: 6
display_port: 8097
display_server: http://localhost
display_winsize: 256
epoch: latest
epoch_count: 1
fineSizeH: 30 [default: 20]
fineSizeW: 20 [default: 32]
gpu_ids: -1 [default: 0]
init_gain: 0.02
init_type: normal
input_nc: 3
ins_max: 4
ins_per: 2
isTrain: True [default: None]
lambda_A: 10.0
lambda_B: 10.0
lambda_ctx: 1.0
lambda_idt: 1.0
loadSizeH: 330 [default: 100]
loadSizeW: 220 [default: 100]
load_iter: 0 [default: 0]
lr: 0.0002
lr_decay_iters: 50
lr_policy: lambda
max_dataset_size: inf
model: insta_gan
n_layers_D: 3
name: jeans2skirt_ccp_instagan [default: experiment_name]
ndf: 64
netD: set
netG: set
ngf: 64
niter: 100
niter_decay: 100
no_dropout: True
no_flip: False
no_html: False
no_lsgan: False
norm: instance
num_threads: 4
output_nc: 3
phase: train
pool_size: 50
print_freq: 100
resize_or_crop: resize_and_crop
save_by_iter: False
save_epoch_freq: 5
save_latest_freq: 5000
serial_batches: False
set_order: decreasing
suffix:
update_html_freq: 1000
verbose: False
----------------- End -------------------
dataset [UnalignedSegDataset] was created
#training images = 167
initialize network with normal
initialize network with normal
initialize network with normal
initialize network with normal
model [InstaGANModel] was created
---------- Networks initialized -------------
[Network G_A] Total number of parameters : 26.815 M
[Network G_B] Total number of parameters : 26.815 M
[Network D_A] Total number of parameters : 9.754 M
[Network D_B] Total number of parameters : 9.754 M
-----------------------------------------------
create web directory ./checkpoints/jeans2skirt_ccp_instagan/web...
Traceback (most recent call last):
File "train.py", line 32, in <module>
model.optimize_parameters()
File "/home/ysn/instagan/models/insta_gan_model.py", line 295, in optimize_parameters
self.real_A_img_sng = self.fake_B_img_sng.detach()
AttributeError: InstaGANModel instance has no attribute 'fake_B_img_sng'
If I increase the fineSize, I get a CUDA memory error.
Any help would be welcome.
Thank you
Hello,
I am trying to train on the jean2skirt dataset and after the following train command:
python train.py --dataroot ./datasets/jeans2skirt_ccp --model insta_gan --name jeans2skirt_ccp_instagan --loadSizeH 330 --loadSizeW 220 --fineSizeH 30 --fineSizeW 20 --gpu_ids -1I get the following error:
If I increase the fineSize, I get a CUDA memory error.
Any help would be welcome.
Thank you