From e19dc78af85e37084187a879fe6e5db4b53a14a2 Mon Sep 17 00:00:00 2001
From: reacher-l <45810596+reacher-l@users.noreply.github.com>
Date: Sat, 10 Apr 2021 19:11:01 +0800
Subject: [PATCH] Update train.py

---
 train.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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,