From fd2ff6616f2df75c741784daebbc69595094257e Mon Sep 17 00:00:00 2001
From: Iridoudou <2534936416@qq.com>
Date: Fri, 3 Sep 2021 15:31:39 +0800
Subject: [PATCH] Support NTU

---
 fit/configs/NTU.json | 4 ++--
 fit/tools/main.py    | 2 +-
 2 files changed, 3 insertions(+), 3 deletions(-)

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))