Method on_train_end

t-few/src/models/EncoderDecoder.py:319–323

(self)

        }

    def on_train_end(self):
        self.save_model(finish=True)

        if self.config.fishmask_mode is not None:
            fishmask_plugin_on_end(self)

    def on_test_start(self):
        # Evaluate model on best metric if training set exists