public-projects/Mouse2Vec
2
0
Fork 0
Code Issues Pull requests Releases Wiki Activity

Updated format of readme

Browse source
This commit is contained in:
Guanhua Zhang 2024-05-07 17:05:50 +02:00
parent b429f9807d
commit dd4a0e2d2e
1 changed files with 1 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
README.md
View file

@ -20,9 +20,7 @@ We recommend to setup a virtual environment using Anaconda. <br>
# Run the code # Run the code
Our code supports training using GPUs or CPUs. It will prioritise GPUs if available (line 45 in main.py). You can also assign a particular card via CUDA_VISIBLE_DEVICES (e.g., the following commands use GPU card no.3). Our code supports training using GPUs or CPUs. It will prioritise GPUs if available (line 45 in main.py). You can also assign a particular card via CUDA_VISIBLE_DEVICES (e.g., the following commands use GPU card no.3).
## Train Mouse2Vec Autoencoder ## Train Mouse2Vec Autoencoder<br>
<br>
Execute Execute
```shell ```shell
CUDA_VISIBLE_DEVICES=3 python main.py --ssl True CUDA_VISIBLE_DEVICES=3 python main.py --ssl True