update

257c722d · Eric.Lee2021 · 8aee7a26 · 257c722d
隐藏空白更改
内联并排

Showing with 1 addition and 2 deletion

train.py train.py +1 -2

未找到文件。
--- a/train.py
+++ b/train.py
@@ -150,8 +150,7 @@ def trainer(ops,f_log):
                optimizer.zero_grad()
                step += 1

-
-            torch.save(model_.state_dict(), ops.model_exp + '{}-model_epoch-{}.pth'.format(ops.model,epoch))
+            torch.save(model_.state_dict(), ops.model_exp + '{}-size-{}-model_epoch-{}.pth'.format(ops.model,ops.img_size[0],epoch))

    except Exception as e:
        print('Exception : ',e) # 打印异常