def test_rmsprop_gpu(self): np.random.seed(1337) model = TinyBobNet() [x.cuda_() for x in model.parameters()] optimizer = optim.RMSprop(model.parameters(), lr=0.0002) train(model, optimizer, steps=1000, gpu=True) evaluate(model, gpu=True)
def test_rmsprop(self): np.random.seed(1337) model = TinyBobNet() optimizer = optim.RMSprop(model.parameters(), lr=0.0002) train(model, X_train, Y_train, optimizer, steps=1000, device=self.device) assert evaluate(model, X_test, Y_test, device=self.device) > 0.95
def test_rmsprop(self): np.random.seed(1337) model = TinyBobNet() optimizer = optim.RMSprop(model.parameters(), lr=0.0002) train(model, optimizer, steps=1000, gpu=self.gpu) evaluate(model, gpu=self.gpu)
def test_rmsprop(self): np.random.seed(1337) model = TinyBobNet() optimizer = optim.RMSprop([model.l1, model.l2], lr=0.0002) train(model, optimizer, steps=1000) evaluate(model)