Update README.md

yahuiwei123 · Dec 20, 2023 · b9ba323 · b9ba323
1 parent 1320533
commit b9ba323
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/README.md b/README.md
@@ -9,7 +9,7 @@ Segment_SNN主要实现了一个利用VGG16和FPN作为backbone，采用encoder-
   + xxx
 ##### 模型训练
 ``
-python train.py --batch_size 8 --step 8 --learning_rate 0.01 --num_epochs 100 -output_size (128, 128)
+python train.py --batch_size 8 --step 8 --learning_rate 0.01 --num_epochs 100 -output_size (480, 480)
 ``
 
 ##### 模型预测