Skip to content

Commit 0e705ba

Browse files
committed
样例执行命令修正
1 parent cdc82e2 commit 0e705ba

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

AdelaiDet/demo_fcos.py

+3-2
Original file line numberDiff line numberDiff line change
@@ -177,11 +177,12 @@ def main(args):
177177

178178
if __name__ == "__main__":
179179
"""
180-
Usage:export KMP_DUPLICATE_LIB_OK=TRUE
180+
Usage:export CUDA_DEVICE_ORDER="PCI_BUS_ID"
181+
export CUDA_VISIBLE_DEVICES="0"
181182
python AdelaiDet/demo_fcos.py --config-file AdelaiDet/R_50_1x.yaml \
182183
--input ./examples/pic1.jpg \
183184
--layer-name proposal_generator.fcos_head.cls_tower.8 \
184-
--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
185186
"""
186187
mp.set_start_method("spawn", force=True)
187188
arguments = get_parser().parse_args()

0 commit comments

Comments
 (0)