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
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
<br>
## Train Mouse2Vec Autoencoder<br>
Execute
```shell
CUDA_VISIBLE_DEVICES=3 python main.py --ssl True