Add DRF Scoliosis1K pipeline and optional wandb logging

2026-03-07 17:49:19 +08:00
parent 654409ff50
commit 51eee70a4b
12 changed files with 1257 additions and 151 deletions
@@ -431,6 +431,7 @@ class BaseModel(MetaModel, nn.Module):
                        model.fix_BN()
                    if result_dict:
                        model.msg_mgr.write_to_tensorboard(result_dict)
+                        model.msg_mgr.write_to_wandb(result_dict)
                    model.msg_mgr.reset_time()
            if model.iteration >= model.engine_cfg['total_iter']:
                break