并行训练代码优化

test_ori.py

@@ -14,6 +14,7 @@ from tools.dataset import get_transform
 from tools.image_joint import merge_imgs
 from configs import trainer_tools
 import yaml
+from datetime import datetime
 
 with open('configs/test.yml', 'r') as f:
     conf = yaml.load(f, Loader=yaml.FullLoader)