zdou0830 commited on
Commit
0443d1a
1 Parent(s): 174cfff
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -49,11 +49,11 @@ weight_file = "MODEL/desco_glip_tiny.pth"
49
 
50
  # update the config options with the config file
51
  # manual override some options
52
- #cfg.local_rank = 0
53
- #cfg.num_gpus = 1
54
  cfg.merge_from_file(config_file)
55
- #cfg.merge_from_list(["MODEL.WEIGHT", weight_file])
56
- #cfg.merge_from_list(["MODEL.DEVICE", "cuda"])
57
 
58
  glip_demo = GLIPDemo(
59
  cfg,
 
49
 
50
  # update the config options with the config file
51
  # manual override some options
52
+ cfg.local_rank = 0
53
+ cfg.num_gpus = 1
54
  cfg.merge_from_file(config_file)
55
+ cfg.merge_from_list(["MODEL.WEIGHT", weight_file])
56
+ cfg.merge_from_list(["MODEL.DEVICE", "cuda"])
57
 
58
  glip_demo = GLIPDemo(
59
  cfg,