图像描述 训练

[249001762]

您好，大佬，请教 example_train_seq2seq.py 中的训练数据问题，
src_max_len = max(len(x["src"]) for x in batch_data)
trg_max_len = max(len(x["trg"]) - 1 for x in batch_data)
我可以在图像描述的训练中使用它吗。src指向到图片地址，然后自己写解析src代码。感谢任何回复

[yaoxiaoyuan]

你可以参考https://github.com/yaoxiaoyuan/mimix/blob/main/example_train_image_caption.py

[249001762]

感谢您！