config.py 306 B

12345678910111213
  1. config = {
  2. 'max_epoch' : 50,
  3. 'log_train' : 100,
  4. 'lr' : 1e-5,
  5. 'starting_epoch' : 0,
  6. 'batch_size' : 1,
  7. 'log_val' : 10,
  8. 'task' : 'abnormal', # "meniscus" and "acl" are the other options
  9. 'weight_decay' : 0.01,
  10. 'patience' : 5,
  11. 'save_model' : 1,
  12. 'exp_name' : 'test'
  13. }