Method move_batch

pretrain_src/model.py:40–41

(self, batch, non_blocking=False)

                       map_location=lambda storage, loc: storage))

    def move_batch(self, batch, non_blocking=False):
        return batch.to(self.device, non_blocking)

    def eval(self):
        self.network.eval()