(self, save_dir, tag, zero_checkpoint)
| 5013 | self.checkpoint_engine.save(saveable_state_dict, save_path) |
| 5014 | |
| 5015 | def _create_checkpoint_file(self, save_dir, tag, zero_checkpoint): |
| 5016 | name_function = (self._get_zero_ckpt_name if zero_checkpoint else self._get_ckpt_name) |
| 5017 | try: |
| 5018 | checkpoint_name = name_function(save_dir, tag) |
| 5019 | path = os.path.dirname(checkpoint_name) |
| 5020 | self.checkpoint_engine.makedirs(path, exist_ok=True) |
| 5021 | except OSError: |
| 5022 | logger.error(f"Failed saving model checkpoint to {save_dir} with tag {tag}") |
| 5023 | return False |
| 5024 | |
| 5025 | return True |
| 5026 | |
| 5027 | def _create_zero_checkpoint_files(self, save_dir, tag): |
| 5028 | success = True |
no test coverage detected