diff --git a/README.md b/README.md index 53fdd49..add03e7 100644 --- a/README.md +++ b/README.md @@ -102,9 +102,27 @@ with the following parameters * ```--save_cycle``` save model, loss, validation every save_cycle epochs, ```default=5``` * ```--weight_decay``` weight_decay for stochastic gradient descent, ```default= 4e5``` -example execution: +example execution with 41.000 input/target images: python3 test_parse.py --input=data/training --target=data/target --content=output --train_tensor_size=2000 --val_tensor_size=1000 --num_train_tensors=20 model_weights=model.pt + +configuration from example execution: + +* ```input_path data/training``` +* ```target_path data/target``` +* ```content_path output``` +* ```batch_size 2000``` +* ```train_tensor_size=20``` +* ```val_tensor_size=1000``` +* ```num_train_tensors=20``` +* ```model_weights=model.pt``` +* ```load_model=True``` +* ```learn_rate=0.0001``` +* ```momentum=0.9``` +* ```weight_decay=400000.0``` +* ```total_epochs=50``` +* ```save_cycle=5``` + ## Results ##