소스 검색

2ndstage-bs

Jazzcharles 8 달 전
부모
커밋
cfcbc3ab1a
1개의 변경된 파일1개의 추가작업 그리고 1개의 파일을 삭제
  1. 1 1
      configs/ovsegmentor/ovsegmentor_pretrain_vit_bert_stage2.yml

+ 1 - 1
configs/ovsegmentor/ovsegmentor_pretrain_vit_bert_stage2.yml

@@ -7,7 +7,7 @@ data:
       root_dir: 's3://GCC/GCC12m/'
       meta_file: '/mnt/petrelfs/xujilan/data/cc12m_100/subsets/cc12m_filtered_subset_pair_debug.csv'
       read_from: petrel
-      batch_size: 4 # 128
+      batch_size: 128
   val:
       root_dir: '/mnt/cache/share/images/val/'