From 1a5646d843349a75a76036db1bd71e891d260e52 Mon Sep 17 00:00:00 2001
From: Iridoudou <2534936416@qq.com>
Date: Sat, 7 Aug 2021 21:32:53 +0800
Subject: [PATCH] update code

---
 fit/configs/UTD_MHAD.json |  2 +-
 fit/tools/main.py         |  3 ---
 fit/tools/train.py        | 21 ---------------------
 3 files changed, 1 insertion(+), 25 deletions(-)

diff --git a/fit/configs/UTD_MHAD.json b/fit/configs/UTD_MHAD.json
index 0e175fb..fc91d77 100644
--- a/fit/configs/UTD_MHAD.json
+++ b/fit/configs/UTD_MHAD.json
@@ -4,7 +4,7 @@
     },
     "TRAIN": {
         "LEARNING_RATE": 2e-2,
-        "MAX_EPOCH": 10,
+        "MAX_EPOCH": 500,
         "WRITE": 1
     },
     "USE_GPU": 1,
diff --git a/fit/tools/main.py b/fit/tools/main.py
index c0021d5..3147ce1 100644
--- a/fit/tools/main.py
+++ b/fit/tools/main.py
@@ -1,16 +1,13 @@
 import torch
-import numpy as np
 from tensorboardX import SummaryWriter
 from easydict import EasyDict as edict
 import time
-import inspect
 import sys
 import os
 import logging
 
 import argparse
 import json
-from tqdm import tqdm
 sys.path.append(os.getcwd())
 from display_utils import display_model
 from smplpytorch.pytorch.smpl_layer import SMPL_Layer
diff --git a/fit/tools/train.py b/fit/tools/train.py
index 3c0a74b..4990177 100644
--- a/fit/tools/train.py
+++ b/fit/tools/train.py
@@ -1,32 +1,11 @@
-from fit.tools.save import save_pic
-import matplotlib as plt
-from matplotlib.pyplot import show
 import torch
-import torch.nn as nn
 import torch.nn.functional as F
-from torch.nn.modules import module
-from torch.optim import lr_scheduler
 import torch.optim as optim
-from torch.utils.data import DataLoader
-import torchvision.datasets as dset
-import torchvision.transforms as T
-import numpy as np
-import scipy.io
-from tensorboardX import SummaryWriter
-from easydict import EasyDict as edict
-import time
-import inspect
 import sys
 import os
-import logging
 
-import argparse
-import json
 from tqdm import tqdm
 sys.path.append(os.getcwd())
-from smplpytorch.pytorch.smpl_layer import SMPL_Layer
-from display_utils import display_model
-from map import mapping
 
    
 class Early_Stop: