To receive notifications about scheduled maintenance, please subscribe to the mailing-list gitlab-operations@sympa.ethz.ch. You can subscribe to the mailing-list at https://sympa.ethz.ch

Commit d60e42b6 authored by zpgeng's avatar zpgeng
Browse files

Update README

parent 8bd351bd
......@@ -18,13 +18,14 @@ config['split'] : True if you want to run a clustered version of the model
Please find the files which are related to our model:
- CNN.CNN.py
- DeepEye.deepeye.py
- DeepEyeRNN.deepeyeRNN.py
- EEGNet.eegNet.py
- InceptionTime.inception.py
- Xception.xception.py
- ConvNet.py
|
| - CNN -- CNN.py
| - DeepEye -- deepeye.py
| - DeepEyeRNN -- deepeyeRNN.py
| - EEGNet -- eegNet.py
| - InceptionTime -- inception.py
| - Xception -- xception.py
| - ConvNet.py
You can find the architechture of our models in these files. For `CNN`, `DeepEye`, `DeepEyeRNN`, `InceptionTime`, and `Xception`, you should tune the parameter by looking into the `ConvNet.py` file and adjust the parameters (e.g. `self.nb_filter`) accordingly.
......@@ -34,9 +35,9 @@ For `EEGNet` model, you should directly look into the `eegNet.py` file and tune
For [ETH Leonhard](https://scicomp.ethz.ch/wiki/Python_on_Leonhard) users, you can follow these steps:
1. Please use the command `module load gcc/6.3.0 python_gpu/3.8.5 hdf5/1.10.1`before training the model.
1. Please use the command `module load gcc/6.3.0 python_gpu/3.8.5 hdf5/1.10.1` before training the model.
2. Edit the data directory and run directory where you saved the datasets.
2. Edit the _data directory_ and __run directory__ where you _saved the datasets_ and __run the code for saving the logs and outputs__.
3. Use the following command to run the model: `bsub -n 10 -W 4:00 -o experiment_[your_model] -R "rusage[mem=5000, ngpus_excl_p=1]" python ../main.py`
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment