[Fix] fix basicvsr train config (#1485)

* fix srcnn * [Fix] fix basicvsr train config
open-mmlab · Nov 30, 2022 · fbdc85f · fbdc85f
1 parent c9149b9
commit fbdc85f
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/configs/basicvsr/basicvsr_2xb4_vimeo90k-bd.py b/configs/basicvsr/basicvsr_2xb4_vimeo90k-bd.py
@@ -52,11 +52,13 @@
         data_prefix=dict(img='BDx4', gt='GT'),
         ann_file='meta_info_Vimeo90K_train_GT.txt',
         depth=2,
+        num_input_frames=7,
         fixed_seq_len=7,
         load_frames_list=dict(img=file_list, gt=file_list),
         pipeline=train_pipeline))
 
 val_dataloader = dict(
+    _delete_=True,
     num_workers=1,
     batch_size=1,
     persistent_workers=False,

diff --git a/configs/basicvsr_pp/basicvsr-pp_c64n7_8xb1-600k_reds4.py b/configs/basicvsr_pp/basicvsr-pp_c64n7_8xb1-600k_reds4.py
@@ -24,7 +24,8 @@
         output_view=(1, -1, 1, 1),
     ))
 
-train_dataloader = dict(num_workers=6, batch_size=1)
+train_dataloader = dict(
+    num_workers=6, batch_size=1, dataset=dict(num_input_frames=30))
 
 train_cfg = dict(
     type='IterBasedTrainLoop', max_iters=600_000, val_interval=5000)