From b4cf4776f118923aa3d2c39e53bc7953d36f091e Mon Sep 17 00:00:00 2001
From: xiaoyuan1996 <745087337@qq.com>
Date: Sat, 6 May 2023 22:31:15 +0800
Subject: [PATCH] update

---
 README.md | 13 ++++++++++++-
 1 file changed, 12 insertions(+), 1 deletion(-)

diff --git a/README.md b/README.md
index 9def802..a472bbe 100644
--- a/README.md
+++ b/README.md
@@ -34,7 +34,18 @@ python scripts/txt2img.py \
 ```
 
 ### Traing
-We will update the train code ASAP.
+Put images of [RSITMD](https://github.com/xiaoyuan1996/AMFMN) in `data/RSITMD/images`, and run with:
+```commandline
+python main.py \
+    -t \
+    --base configs/lammbda/RSITMD.yaml \
+    --gpus 1 \
+    --scale_lr False \
+    --num_nodes 1 \
+    --check_val_every_n_epoch 10 \
+    --finetune_from './last-pruned.ckpt'
+```
+
 
 ## Examples
 **Caption:** `Some boats drived in the sea.`