We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent cdc82e2 commit 0e705baCopy full SHA for 0e705ba
AdelaiDet/demo_fcos.py
@@ -177,11 +177,12 @@ def main(args):
177
178
if __name__ == "__main__":
179
"""
180
- Usage:export KMP_DUPLICATE_LIB_OK=TRUE
+ Usage:export CUDA_DEVICE_ORDER="PCI_BUS_ID"
181
+ export CUDA_VISIBLE_DEVICES="0"
182
python AdelaiDet/demo_fcos.py --config-file AdelaiDet/R_50_1x.yaml \
183
--input ./examples/pic1.jpg \
184
--layer-name proposal_generator.fcos_head.cls_tower.8 \
- --opts MODEL.WEIGHTS /sdb/tmp/pretrained_model/fcos_R_50_1x.pth MODEL.DEVICE cuda
185
+ --opts MODEL.WEIGHTS /path/to/fcos_R_50_1x.pth MODEL.DEVICE cuda
186
187
mp.set_start_method("spawn", force=True)
188
arguments = get_parser().parse_args()
0 commit comments