You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
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
The text was updated successfully, but these errors were encountered:
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:
If I increase the fineSize, I get a CUDA memory error.
Any help would be welcome.
Thank you
The text was updated successfully, but these errors were encountered: