out_dir / checkpoint-500
jrd971000's picture
End of training
3fba716