SCNet-large pretrain model

starrytong · Sep 23, 2024 · 6236f8c · 6236f8c
1 parent 4a414b3
commit 6236f8c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/README.md b/README.md
@@ -41,8 +41,8 @@ The large version is now available.
 
 You need to modify two parameters in the config.yaml file. 
 ```bash
-dim = [4, 64, 128, 256]
-band_SR = [0.225, 0.372, 0.403]
+dims: [4, 64, 128, 256]
+band_SR: [0.225, 0.372, 0.403]
 ```
 I have performed normalization on the model's input during training, which helps in stabilizing the training process (no code modifications are needed during inference).