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)
