KevinQHLin commited on
Commit
b3fb703
·
1 Parent(s): ef19d17

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -11,7 +11,7 @@ from run_on_video import clip, vid2clip, txt2clip
11
  parser = argparse.ArgumentParser(description='')
12
  parser.add_argument('--save_dir', type=str, default='./tmp')
13
  parser.add_argument('--resume', type=str, default='./results/omni/model_best.ckpt')
14
- parser.add_argument("--gpu_id", type=int, default=2)
15
  args = parser.parse_args()
16
  os.environ["CUDA_VISIBLE_DEVICES"] = str(args.gpu_id)
17
 
 
11
  parser = argparse.ArgumentParser(description='')
12
  parser.add_argument('--save_dir', type=str, default='./tmp')
13
  parser.add_argument('--resume', type=str, default='./results/omni/model_best.ckpt')
14
+ parser.add_argument("--gpu_id", type=int, default=0)
15
  args = parser.parse_args()
16
  os.environ["CUDA_VISIBLE_DEVICES"] = str(args.gpu_id)
17