From a5a6d842b7773dc4298d3f83e277457d2ec6a29c Mon Sep 17 00:00:00 2001
From: pp00704831 <57279608+pp00704831@users.noreply.github.com>
Date: Sat, 16 Jul 2022 21:35:24 +0800
Subject: [PATCH] Update README.md

---
 README.md | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/README.md b/README.md
index c2f0ea2..27065ce 100644
--- a/README.md
+++ b/README.md
@@ -24,12 +24,12 @@ For example: './datasets/GoPro/train/blur/\*\*/\*.png'
 **We train our Stripformer in two stages:** </br>
 **1) We pre-train Stripformer for 3000 epochs on patch size 256x256. Please run the following commands.** </br>
 ```
-python pretrained.py
+python train_Stripformer_pretrained.py
 ```
 
 **2) After stage 1, we keep training Stripformer for 1000 epochs on patch size 512x512. Please run the following commands.** </br>
 ```
-python train.py
+python train_Stripformer_gopro.py
 ```
 
 ## Testing