From 256bbc42375dfc4118821bd94e20b945b7014d41 Mon Sep 17 00:00:00 2001
From: Hobin Ryu <hobincar@kaist.ac.kr>
Date: Wed, 11 Dec 2019 00:39:35 +0900
Subject: [PATCH] Update train_stage2.py

refers #20
---
 configs/train_stage2.py | 8 --------
 1 file changed, 8 deletions(-)

diff --git a/configs/train_stage2.py b/configs/train_stage2.py
index 9f935ac..e35a50b 100644
--- a/configs/train_stage2.py
+++ b/configs/train_stage2.py
@@ -18,10 +18,6 @@ class VocabConfig:
 
 
 class MSVDLoaderConfig:
-    n_train = 1200
-    n_val = 100
-    n_test = 670
-
     total_caption_fpath = "data/MSVD/metadata/MSR Video Description Corpus.csv"
     train_caption_fpath = "data/MSVD/metadata/train.csv"
     val_caption_fpath = "data/MSVD/metadata/val.csv"
@@ -39,10 +35,6 @@ class MSVDLoaderConfig:
 
 
 class MSRVTTLoaderConfig:
-    n_train = 5175
-    n_val = 398
-    n_test = 2354
-
     total_caption_fpath = "data/MSR-VTT/metadata/total.json"
     train_caption_fpath = "data/MSR-VTT/metadata/train.json"
     val_caption_fpath = "data/MSR-VTT/metadata/val.json"