Spaces:
Runtime error
Runtime error
desco
Browse files
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 |
-
|
53 |
-
|
54 |
cfg.merge_from_file(config_file)
|
55 |
-
|
56 |
-
|
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,
|