File size: 589 Bytes
131da64
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
# @package _global_
  
data:
  keep_hf_dataset_in_memory: true
  aggressive_aug: false
  n_duplicate_train: null
  n_duplicate_val: null

  tokenize_vqvae_in_dataloader: false
  enable_cuda_in_tensordict_collate: false
  force_mp_spawn: false
  keep_tensordict_on_disk: false
  move_tensordict_to_shm: false

  fid_dataset: datacomp1b_8_magvit_val
  image_data_train: null
  image_data_val: null
  data_dir_train: ${data.data_dir_val}
  data_dir_val:
    - dir: ${oc.env:DIFFUSION_DATA_DIR}/tokens/scratch_ssd_tokens/datacomp1b_8_magvit_val
      weight: -1
      name: ${data.fid_dataset}