diff --git a/train.py b/train.py
index 566798b..09d28f9 100644
--- 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,