add gaitedge training code

2022-07-17 13:47:50 +08:00
parent 4205c5f283
commit b183455eb8
17 changed files with 814 additions and 11 deletions
@@ -8,7 +8,6 @@ data_cfg:
  test_dataset_name: GREW

 evaluator_cfg:
-  enable_distributed: true
  enable_float16: false 
  restore_ckpt_strict: true
  restore_hint: 250000
@@ -53,7 +52,6 @@ scheduler_cfg:
  scheduler: MultiStepLR

 trainer_cfg:
-  enable_distributed: true
  enable_float16: true
  with_test: false
  log_iter: 100