how to change the epoch of train?

muzairkhattak / multimodal-prompt-learning

[CVPR 2023] Official repository of paper titled "MaPLe: Multi-modal Prompt Learning".

MIT License

619 stars 43 forks source link

Closed SHIBOYA closed 1 year ago

SHIBOYA commented 1 year ago

when i try to change the epoch of the model, i can only find the epoch in the test.sh. how to change the epoch of train?

muzairkhattak commented 1 year ago

Hi @SHIBOYA!

You can change the number of epochs in the config file (check here) using the MAX_EPOCH argument.

Thank you!