Method training_end

tasks/base_task.py:273–274

(self, *args, **kwargs)

        return model

    def training_end(self, *args, **kwargs):
        return None

    def init_ddp_connection(self, proc_rank, world_size):
        set_hparams(print_hparams=False)