Arpit16112's picture
Adding the complete NST trained model code
0c3960b
Raw
History Blame Contribute Delete
354 Bytes
content_dir: ./content_data_examples
style_dir: ./style_data_examples
vgg: vgg_normalised.pth
experiment: experiment1
final_size: 256
content_size: 512
style_size: 512
crop: True
batch_size: 4
lr: 0.0001
lr_decay: 5e-05
epochs: 10
content_weight: 1.0
style_weight: 5
log_interval: 1
save_interval: 2
resume: False
decoder_path: None
optimizer_path: None