(self, val_loss, model=None)
| 174 | self.model_path = "hdd_data/prepare_dataset/model/model.pt" |
| 175 | |
| 176 | def __call__(self, val_loss, model=None): |
| 177 | |
| 178 | score = -val_loss |
| 179 | |
| 180 | if self.best_score is None: |
| 181 | self.best_score = score |
| 182 | # self.save_checkpoint(val_loss, model) |
| 183 | elif score < self.best_score + self.delta: |
| 184 | self.counter += 1 |
| 185 | if self.verbose: |
| 186 | print(f'EarlyStopping counter: {self.counter} out of {self.patience}') |
| 187 | if self.counter >= self.patience: |
| 188 | self.early_stop = True |
| 189 | return True |
| 190 | else: |
| 191 | self.best_score = score |
| 192 | # self.save_checkpoint(val_loss, model) |
| 193 | self.counter = 0 |
| 194 | |
| 195 | def save_checkpoint(self, val_loss, model): |
| 196 | '''Saves model when validation loss decrease.''' |
nothing calls this directly
no outgoing calls
no test coverage detected