Support NTU

2021-09-03 15:31:39 +08:00
parent 52603950d6
commit fd2ff6616f
2 changed files with 3 additions and 3 deletions
--- a/fit/tools/main.py
+++ b/fit/tools/main.py
@ -115,7 +115,7 @@ if __name__ == "__main__":
            logger.info("avg_loss:{:.4f}".format(meters.avg))

            save_params(res, file, logger, args.dataset_name)
-            save_pic(res,smpl_layer,file,logger,args.dataset_name,target)
+            # save_pic(res,smpl_layer,file,logger,args.dataset_name,target)
            
            torch.cuda.empty_cache()
    logger.info("Fitting finished! Average loss:     {:.9f}".format(meters.avg))