Arpit16112's picture
Adding the complete NST trained model code
0c3960b
Raw
History Blame Contribute Delete
425 Bytes
content_dir: ./content_dataset
style_dir: ./style_dataset
vgg: vgg_normalised.pth
experiment: final_training
final_size: 512
content_size: 512
style_size: 512
crop: True
resume: True
lr: 0.0001
lr_decay: 5e-05
batch_size: 8
epochs: 200
style_weight: 10
content_weight: 1.0
log_interval: 1
save_interval: 20
decoder_path: experiment/final_training/decoder_160.pth
optimizer_path: experiment/final_training/optimizer_160.pth