pere commited on
Commit
422b7cb
1 Parent(s): 5a96a32
Files changed (1) hide show
  1. finetune_categorisation_xxl.gin +1 -1
finetune_categorisation_xxl.gin CHANGED
@@ -17,7 +17,7 @@ TRAIN_STEPS = %gin.REQUIRED #1_010_000 # 1000000 pre-trained steps + 10000 fine
17
  USE_CACHED_TASKS = False
18
  DROPOUT_RATE = 0.1
19
  RANDOM_SEED = 0
20
- BATCH_SIZE = 64
21
 
22
  #Fixing a small error
23
  infer_eval/utils.DatasetConfig:
 
17
  USE_CACHED_TASKS = False
18
  DROPOUT_RATE = 0.1
19
  RANDOM_SEED = 0
20
+ BATCH_SIZE = 32
21
 
22
  #Fixing a small error
23
  infer_eval/utils.DatasetConfig: