Skip to content
Discussion options

You must be logged in to vote

Thanks to @tchaton on the slack community I solved the issue overriding the ModelCheckpoint class.
In the on_train_epoch_end I've added a new check that follow the above conditions, as such:

class WGANModelCheckpoint(ModelCheckpoint):
    def __init__(self,
                 dirpath: Optional[Union[str, Path]] = None,
                 filename: Optional[str] = None,
                 monitor: Optional[str] = None,
                 verbose: bool = False,
                 save_last: Optional[bool] = None,
                 save_top_k: int = 1,
                 save_weights_only: bool = False,
                 mode: str = "min",
                 auto_insert_metric_name: bool = True,
           …

Replies: 1 comment

Comment options

You must be logged in to vote
0 replies
Answer selected by SalvatoreCognetta
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment