Method __init__

examples/bgrl/train.py:32–36

(self, args)

class ModelTrainer:

    def __init__(self, args):
        self._args = args

        self._init()
        self.writer = SummaryWriter(log_dir="saved/BGRL_dataset({})".format(args.name))

    def _init(self):
        args = self._args