Merge pull request nitrain#58 from miguelvr/fix-readme

Fix to README.md to match recent changes
KunWangV · Oct 8, 2017 · bba5ccf · bba5ccf
2 parents 576160a + 74fb6a7
commit bba5ccf
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/README.md b/README.md
@@ -48,7 +48,7 @@ trainer.compile(loss='nll_loss',
                 optimizer='adadelta')
 
 trainer.fit(x_train, y_train, 
-            validation_data=(x_test, y_test),
+            val_data=(x_test, y_test),
             num_epoch=20, 
             batch_size=128,
             verbose=1)
@@ -120,7 +120,7 @@ train_loader = DataLoader(train_dataset, batch_size=32)
 val_dataset = TensorDataset(x_val, y_val)
 val_loader = DataLoader(val_dataset, batch_size=32)
 
-model.fit_loader(loader, val_loader=val_loader, nb_epoch=100)
+trainer.fit_loader(loader, val_loader=val_loader, num_epoch=100)
 ```
 
 ## Utility Functions