has been renamed to save_16bit_model, keeping this around for backwards compatibility
(self, save_dir, save_filename="pytorch_model.bin")
| 5479 | return state_dict |
| 5480 | |
| 5481 | def save_fp16_model(self, save_dir, save_filename="pytorch_model.bin"): |
| 5482 | """has been renamed to save_16bit_model, keeping this around for backwards |
| 5483 | compatibility""" |
| 5484 | return self.save_16bit_model(save_dir, save_filename) |
| 5485 | |
| 5486 | def save_16bit_model(self, save_dir, save_filename="pytorch_model.bin", exclude_frozen_parameters=False): |
| 5487 | """ |
nothing calls this directly
no test coverage detected