output_dir / checkpoint-2000
Mahmoud7's picture
End of training
b754cad