Added public link to docuumentation and small fix

README.rst

@@ -2,9 +2,9 @@
 .. Mon 03 Jul 2017 11:50:06 CEST
 
 .. image:: http://img.shields.io/badge/docs-stable-yellow.svg
-   :target: http://beatubulatest.lab.idiap.ch/private/docs/bob/bob.paper.mccnn.tifs2018/stable/index.html
+   :target: http://idiap.ch/software/docs/bob/bob.paper.mccnn.tifs2018/stable/index.html
 .. image:: http://img.shields.io/badge/docs-latest-orange.svg
-   :target: http://beatubulatest.lab.idiap.ch/private/docs/bob/bob.paper.mccnn.tifs2018/master/index.html
+   :target: http://idiap.ch/software/docs/bob/bob.paper.mccnn.tifs2018/master/index.html
 .. image:: https://gitlab.idiap.ch/bob/bob.paper.mccnn.tifs2018/badges/master/build.svg
    :target: https://gitlab.idiap.ch/bob/bob.paper.mccnn.tifs2018/commits/master
 .. image:: https://gitlab.idiap.ch/bob/bob.paper.mccnn.tifs2018/badges/master/coverage.svg
@@ -80,4 +80,4 @@ development `mailing list`_.
 .. _mailing list: https://www.idiap.ch/software/bob/discuss
 .. _bob package development: https://www.idiap.ch/software/bob/docs/bob/bob.extension/master/
 .. _conda: https://conda.io
-.. _install conda: https://conda.io/docs/install/quick.html#linux-miniconda-install
+.. _install conda: https://conda.io/docs/install/quick.html#linux-miniconda-install
\ No newline at end of file

bob/paper/mccnn/tifs2018/trainer_configs/wmca_mccnn.py

@@ -70,7 +70,9 @@ SELECTED_CHANNELS = [0, 1, 2, 3]
 
 img_transform_train = transforms.Compose([ChannelSelect(selected_channels=SELECTED_CHANNELS), RandomHorizontalFlipImage(p=0.5), transforms.ToTensor()])
 
-dataset = DataFolder(data_folder=data_folder_train,
+dataset={}
+
+dataset['train'] = DataFolder(data_folder=data_folder_train,
                      transform=img_transform_train,
                      extension='.hdf5',
                      bob_hldi_instance=bob_hldi_instance_train,