diff --git a/README.md b/README.md index ec03ce6..04ca5d9 100644 --- a/README.md +++ b/README.md @@ -20,9 +20,7 @@ We recommend to setup a virtual environment using Anaconda.
# 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 -
- +## Train Mouse2Vec Autoencoder
Execute ```shell CUDA_VISIBLE_DEVICES=3 python main.py --ssl True