Add code of GaitBase (#115)

* add resnet9 backbone and regular da ops * add gait3d config * fix invalid path CASIA-B* in windows * add gaitbase config for all datasets * rm unused OpenGait transform
2023-03-20 14:59:08 +08:00
parent 9b74b39f80
commit 638c657763
16 changed files with 656 additions and 161 deletions
@@ -427,7 +427,8 @@ class BaseModel(MetaModel, nn.Module):
                    model.train()
                    if model.cfgs['trainer_cfg']['fix_BN']:
                        model.fix_BN()
-                    model.msg_mgr.write_to_tensorboard(result_dict)
+                    if result_dict:
+                        model.msg_mgr.write_to_tensorboard(result_dict)
                    model.msg_mgr.reset_time()
            if model.iteration >= model.engine_cfg['total_iter']:
                break