- Clone original AoANet.
- Put
AttModel.py
andCaptionModel.py
in/Path/to/AoANet/models/
, putopts.py
in/Path/to/AoANet/
. - Follow the instructions in original AoANet to train/inference/evaluate your model.
- For evaluation, add
--eval_lvl N
to the inference script to generate captions on levelN
.