diff --git a/README.md b/README.md
index 99ebbd5..b343cf6 100644
--- a/README.md
+++ b/README.md
@@ -97,7 +97,7 @@ python -m torch.distributed.launch --nproc_per_node=1 --master_port=48798 main_f
 ```
 
 
-### Load the model and weights
+### Load the model and weights (if you want to call the model in your code)
 
 ```
 import torch