modify Early_Stop

2021-08-09 11:21:18 +08:00
parent c3dc13d2c8
commit 44d1610296
4 changed files with 4 additions and 4 deletions
--- a/fit/tools/main.py
+++ b/fit/tools/main.py
@ -12,7 +12,7 @@ sys.path.append(os.getcwd())
 from smplpytorch.pytorch.smpl_layer import SMPL_Layer
 from train import train
 from transform import transform
-from save import save_pic,save_params
+from save import save_params
 from load import load
 torch.backends.cudnn.benchmark=True