Hi @mzbac Thanks for the great repo. I want to do prediction on a test dataset after fine tuning. I can see do_prediction arg in the code but it has not been initialized in the args (just do-train is there).
that would be great if you could update the readme how to use this argument
