Upload 10 files

app.py

@@ -43,7 +43,8 @@ from data_utils.lower_body import part2full, pred2poses, poses2pred, poses2poses
 from visualise.rendering import RenderTool
 
 global device
-device = 'cpu'
+is_cuda = torch.cuda.is_available()
+device = torch.device("cuda" if is_cuda else "cpu")
 
 
 def init_model(model_name, model_path, args, config):
@@ -120,6 +121,7 @@ global_orient = torch.tensor([3.0747, -0.0158, -0.0152])
 
 parser = parse_args()
 args = parser.parse_args()
+args.gpu = device
 
 RUN_MODE = "local"
 if RUN_MODE != "local":

trainer/options.py

@@ -2,7 +2,7 @@ from argparse import ArgumentParser
 
 def parse_args():
     parser = ArgumentParser()
-    parser.add_argument('--gpu', default='cpu', type=str)
+    parser.add_argument('--gpu', default='0', type=str)
     parser.add_argument('--save_dir', default='experiments', type=str)
     parser.add_argument('--exp_name', default='smplx_S2G', type=str)
     parser.add_argument('--speakers', nargs='+')