Commit 1399808f authored by Jean Ibarz's avatar Jean Ibarz
Browse files

Update train_model.py

parent 807c8bd5
......@@ -27,13 +27,13 @@ exp_config = ExperimentConfiguration({'n_augment': 1,
'n_epochs': 500,
'n_iters': 1,
'batch_size': 512,
'learning_rate': 0.0001,
'learning_rate': 0.0001, # may use 0.001 for model 'left_center_right'
'model_name': 'one_sided',
# possible values: {'default', 'left_center_right', 'one_sided'}
'model_config': {
'random_scale': {
'minval': -10,
'maxval': 10,
'minval': -1,
'maxval': 1,
},
'random_shift': {
'minval': 0,
......
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment