update code

fit/tools/main.py

@@ -1,5 +1,5 @@
-import numpy as np
 import torch
+import numpy as np
 from tensorboardX import SummaryWriter
 from easydict import EasyDict as edict
 import time

fit/tools/save.py

@@ -6,8 +6,8 @@ import numpy as np
 import pickle
 
 sys.path.append(os.getcwd())
-from display_utils import display_model
 from label import get_label
+from display_utils import display_model
 
 
 def create_dir_not_exist(path):

fit/tools/train.py

@@ -95,6 +95,6 @@ def train(smpl_layer, target,
             writer.add_scalar('learning_rate', float(
                 optimizer.state_dict()['param_groups'][0]['lr']), epoch)
 
-    
-    logger.info('Train ended, min_loss = {:.9f}'.format(float(early_stop.min_loss)))
+    logger.info('Train ended, min_loss = {:.9f}'.format(
+        float(early_stop.min_loss)))
     return res