Add a check on data before fit.

Author: benjamin-work

inferno/net.py

@@ -214,6 +214,9 @@ def initialize(self):
         self.initialized_ = True
         return self
 
+    def check_data(self, *data):
+        pass
+
     def validation_step(self, xi, yi):
         xi, yi = Variable(xi), Variable(yi)
         self.module_.eval()
@@ -254,6 +257,7 @@ def fit_loop(self, X, y):
             self.notify('on_epoch_end', X=X, y=y)
 
     def fit(self, X, y, **fit_params):
+        self.check_data(X, y)
         if self.cold_start or not hasattr(self, 'initialized_'):
             self.initialize()