best_model_epoch_150.pth 2.5 MB