Arpit16112's picture
Adding the complete NST trained model code
0c3960b
Raw
History Blame Contribute Delete
423 Bytes
content_dir: ./content_data_examples
style_dir: ./style_data_examples
vgg: vgg_normalised.pth
experiment: experiment3
final_size: 512
content_size: 512
style_size: 512
crop: True
batch_size: 8
lr: 0.0001
lr_decay: 5e-05
epochs: 6
content_weight: 1.0
style_weight: 10.0
log_interval: 1
save_interval: 2
resume: True
decoder_path: experiment/experiment2/decoder_10.pth
optimizer_path: experiment/experiment2/optimizer_10.pth