WIP: Generic trainer

• Adds a generic trainer class and trainer script
• Moved the model specific stuff to config files; example config added
• uses .to(device) throughout

bob/learn/pytorch/scripts/train_generic.py

@@ -64,7 +64,7 @@ def main(user_input=None):
 
   prog = os.path.basename(sys.argv[0])
   completions = dict(prog=prog, version=version,)
-  args = docopt(__doc__ % completions,argv=arguments,version='Train a FASNet (%s)' % version,)
+  args = docopt(__doc__ % completions,argv=arguments,version='Train a Generic Network  (%s)' % version,)
 
   # load configuration file
   configuration = load([os.path.join(args['<configuration>'])])
@@ -138,6 +138,8 @@ def main(user_input=None):
   else:
     logger.error("Please provide a dataset in your configuration file !")
     sys.exit()
+
+  assert(hasattr(configuration, 'optimizer'))
   
   # train the network
   if hasattr(configuration, 'network'):