Method cuda

nnet/py_factory.py:68–69

(self)

            raise ValueError("unknown optimizer")

    def cuda(self):
        self.model.cuda()

    def train_mode(self):
        self.network.train()