diff --git a/aimodel/src/encoderonly_test_rainfall.py b/aimodel/src/encoderonly_test_rainfall.py
index ad1811d..c6d4355 100755
--- a/aimodel/src/encoderonly_test_rainfall.py
+++ b/aimodel/src/encoderonly_test_rainfall.py
@@ -49,7 +49,8 @@ dataset_train, dataset_validate = dataset_encoderonly(
 	dirpath_input=DIRPATH_INPUT,
 	filepath_heightmap=PATH_HEIGHTMAP,
 	batch_size=BATCH_SIZE,
-	windowsize=WINDOW_SIZE
+	windowsize=WINDOW_SIZE,
+	rainfall_scale_up=2
 )