From 1f4b0778ab7caab4cad897584b09644cc14bd1d3 Mon Sep 17 00:00:00 2001
From: rmaphoh <rmaphoh@ucl.ac.uk>
Date: Mon, 20 Feb 2023 21:31:18 +0000
Subject: [PATCH] update_README

---
 README.md | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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