fix gpt not save issue.

main
RVC-Boss 2 years ago committed by GitHub
parent ecb4b23fc3
commit f0cfe39708
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

@ -44,9 +44,8 @@ class my_model_ckpt(ModelCheckpoint):
self.config = config self.config = config
def on_train_epoch_end(self, trainer, pl_module): def on_train_epoch_end(self, trainer, pl_module):
if not self._should_skip_saving_checkpoint( # if not self._should_skip_saving_checkpoint(trainer) and self._should_save_on_train_epoch_end(trainer):
trainer if self._should_save_on_train_epoch_end(trainer):
) and self._should_save_on_train_epoch_end(trainer):
monitor_candidates = self._monitor_candidates(trainer) monitor_candidates = self._monitor_candidates(trainer)
if ( if (
self._every_n_epochs >= 1 self._every_n_epochs >= 1

Loading…
Cancel
Save