1344dee35dbacaaaaabdaf452f0dfe74e3ab50e4,torchsample/modules/example.py,,,#,35

Before Change

s = time.time()
model.fit(torch.from_numpy(x_train[:10000]), 
          torch.from_numpy(y_train[:10000]), 
          val_data=(torch.from_numpy(x_test), torch.from_numpy(y_test)),
          nb_epoch=2, 
          batch_size=128)
e = time.time()

After Change

// evaluation loop
val_loss = model.evaluate(x_test, y_test)
print("Val Loss: " , val_loss)

// prediction and manual evaluation
from torch.autograd import Variable
y_pred = model.predict(x_test)

Instances