From aa959ee65f76c6810baf126dbecb65f27ffdd512 Mon Sep 17 00:00:00 2001
From: fatih <34196005+fcakyon@users.noreply.github.com>
Date: Fri, 12 Aug 2022 23:55:00 +0300
Subject: [PATCH] update readme (#5)

---
 README.md | 28 ++++++++++++++--------------
 1 file changed, 14 insertions(+), 14 deletions(-)

diff --git a/README.md b/README.md
index e593fc9..ffc93fe 100644
--- a/README.md
+++ b/README.md
@@ -101,13 +101,13 @@ datamodule = VideoDataModule(
     train_dataset_multiplier=1,
     batch_size=4,
     num_workers=4,
-    video_timesteps=8,
-    video_crop_size=224,
-    video_means=backbone.mean,
-    video_stds=backbone.std,
-    video_min_short_side_scale=256,
-    video_max_short_side_scale=320,
-    video_horizontal_flip_p=0.5,
+    num_timesteps=8,
+    preprocess_input_size=224,
+    preprocess_means=backbone.mean,
+    preprocess_stds=backbone.std,
+    preprocess_min_short_side_scale=256,
+    preprocess_max_short_side_scale=320,
+    preprocess_horizontal_flip_p=0.5,
 )
 
 head = LinearHead(hidden_size=neck.num_features, num_classes=datamodule.num_classes)
@@ -143,13 +143,13 @@ datamodule = VideoDataModule(
     train_dataset_multiplier=1,
     batch_size=4,
     num_workers=4,
-    video_timesteps=8,
-    video_crop_size=224,
-    video_means=backbone.mean,
-    video_stds=backbone.std,
-    video_min_short_side_scale=256,
-    video_max_short_side_scale=320,
-    video_horizontal_flip_p=0.5,
+    num_timesteps=8,
+    preprocess_input_size=224,
+    preprocess_means=backbone.mean,
+    preprocess_stds=backbone.std,
+    preprocess_min_short_side_scale=256,
+    preprocess_max_short_side_scale=320,
+    preprocess_horizontal_flip_p=0.5,
 )
 
 head = LinearHead(hidden_size=neck.num_features, num_classes=datamodule.num_classes)