magic
copied
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
Readme
Files and versions
15 lines
458 B
15 lines
458 B
2 years ago
|
#!/bin/bash
|
||
|
|
||
|
# lm_model:
|
||
|
# 1. cambridgeltl/magic_mscoco
|
||
|
# 2. cambridgeltl/magic_flickr30k
|
||
|
CUDA_VISIBLE_DEVICES=1 python run.py \
|
||
|
--beam_size 1 \
|
||
|
--target_seq_length 16 \
|
||
|
--reset_context_delta \
|
||
|
--lm_model cambridgeltl/magic_mscoco \
|
||
|
--test_image_prefix_path ../data/mscoco/test_images \
|
||
|
--test_path ../data/mscoco/mscoco_test.json \
|
||
|
--save_path_prefix ../inference_result/mscoco/baselines/ \
|
||
|
--save_name zerocap_result.json
|