Update train.py

reacher-l · Apr 10, 2021 · e19dc78 · e19dc78
1 parent 2fcde51
commit e19dc78
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/train.py b/train.py
@@ -243,7 +243,7 @@ def create_model(ema=False):
                 best_AA = AA
                 best_loss = epoch_loss
                 torch.save({
-                    'name': 'resnest50_lovasz_edge_rotate',
+                    'name': 'baseline+cosine+ms2+ema+uniformsample',
                     'epoch': epoch + 1,
                     'state_dict': ema_model.state_dict(),
                     'best_miou': best_miou,