support multi loader for train

Author: gemfield

deepvac/core/deepvac.py

@@ -402,7 +402,8 @@ def initOptimizer(self):
     def initTrainLoader(self):
         if self.config.train_loader is None:
             LOG.logE("You must set config.core.train_loader in config.py, or reimplement initTrainLoader() API in your DeepvacTrain subclass {}.".format(self.name()), exit=True)
-        LOG.logI("You set config.core.train_loader to {} in config.py".format(self.config.train_loader)) 
+        LOG.logI("You set config.core.train_loader to {} in config.py".format(self.config.train_loader))
+        self.config.loader = self.config.train_loader
 
     def initValLoader(self):
         if self.config.val_loader is None:
@@ -554,6 +555,8 @@ def train(self):
         self.config.net.train()
         self.initStepAndSaveNumber()
         self.initTickTock()
+        #for multi loader user case
+        self.config.loader = self.config.train_loader
         self.preEpoch()
         self._preEpoch()
         self.doIterTick()

deepvac/datasets/file_line.py

@@ -72,4 +72,4 @@ def __getitem__(self, index):
         if self.config.composer is not None:
             sample, label = self.config.composer([sample, label])
 
-        return sample, label
+        return sample, label