logo
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

#!/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