diff --git a/README.md b/README.md index c6a24a5..4244914 100644 --- a/README.md +++ b/README.md @@ -43,7 +43,7 @@ CUDA_VISIBLE_DEVICES=0 python3 main.py --epochs 70 --batch_size 16 ``` ## Testing Testing is done in two steps. First, is to run test.py script, which saves the prediction outputs in a folder. -Download the pre-trained weights of the transformer-based models on the datasets and run the below commands. +Download the pre-trained weights of the transformer-based models on the datasets and run the below command. ```bash CUDA_VISIBLE_DEVICES=0 python test.py --path_to_model experiments/exp_1/checkpoints/model_best.pth.tar --output_folder experiments/exp_1/test/ --data_folder test ```