diff --git a/fit/configs/NTU.json b/fit/configs/NTU.json index ca5bc81..32a73b5 100644 --- a/fit/configs/NTU.json +++ b/fit/configs/NTU.json @@ -7,12 +7,12 @@ "MAX_EPOCH": 1000, "WRITE": 10, "OPTIMIZE_SCALE":0, - "OPTIMIZE_SHAPE":1 + "OPTIMIZE_SHAPE":0 }, "USE_GPU": 1, "DATASET": { "NAME": "NTU", - "PATH": "../NTU RGB+D/skeleton_npy", + "PATH": "../NTU RGB+D/result", "TARGET_PATH": "", "DATA_MAP": [ [ diff --git a/fit/tools/main.py b/fit/tools/main.py index d76e175..8eadb3a 100644 --- 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))