diffusion_baseline / configs /experiments.yaml
jyliuAI's picture
Add files using upload-large-folder tool
784f572 verified
Raw History Blame Contribute Delete
2.97 kB
step_grids:
many: [1, 2, 4, 8, 16, 32, 1024]
few: [1, 2, 4, 8, 16, 32]
fixed_1024: [1024]
models:
flm:
category: many
environment: dlb-flm
adapter: flm
source: flm
datasets:
lm1b: {status: supported, provenance: official, train_recipe: flm_lm1b_official_ckpt}
owt: {status: supported, provenance: official, train_recipe: flm_owt_official_ckpt}
langflow:
category: many
environment: dlb-langflow
adapter: langflow
source: langflow
datasets:
lm1b: {status: supported, provenance: official}
owt: {status: supported, provenance: official}
duo:
category: many
environment: dlb-duo
adapter: duo
source: duo
datasets:
lm1b: {status: supported, provenance: reference_reproduction}
owt: {status: supported, provenance: official}
mdlm:
category: many
environment: dlb-mdlm
adapter: mdlm
source: mdlm
datasets:
lm1b: {status: supported, provenance: reference_reproduction}
owt: {status: supported, provenance: official}
candi:
category: many
environment: dlb-candi
adapter: candi
source: candi
datasets:
lm1b: {status: supported, provenance: reference_reproduction}
owt: {status: supported, provenance: reference_reproduction, train_recipe: candi_owt}
rdlm:
category: many
step_override: [1000, 1024]
environment: dlb-rdlm
adapter: rdlm
source: rdlm
datasets:
lm1b: {status: supported, provenance: official}
owt:
status: unsupported
reason: The official RDLM release and paper do not provide an OWT model.
fmlm:
category: few
environment: dlb-flm
adapter: flm
source: flm
datasets:
lm1b: {status: supported, provenance: official, train_recipe: fmlm_lm1b_official_ckpt}
owt: {status: supported, provenance: official, train_recipe: fmlm_owt_official_ckpt}
duo_dcd:
category: few
environment: dlb-duo
adapter: duo
source: duo
datasets:
lm1b: {status: supported, provenance: reference_reproduction, train_recipe: duo_dcd_lm1b}
owt: {status: supported, provenance: official}
duo_di4c:
category: few
environment: dlb-di4c
adapter: di4c
source: di4c
datasets:
lm1b: {status: supported, provenance: reference_reproduction, train_recipe: duo_di4c_lm1b}
owt: {status: supported, provenance: reference_reproduction, train_recipe: duo_di4c_owt}
mdlm_sdtt:
category: few
environment: dlb-sdtt
adapter: sdtt
source: sdtt
datasets:
lm1b: {status: supported, provenance: reference_reproduction, train_recipe: mdlm_sdtt_lm1b}
owt: {status: supported, provenance: official}
mdlm_di4c:
category: few
environment: dlb-di4c
adapter: di4c
source: di4c
datasets:
lm1b: {status: supported, provenance: reference_reproduction, train_recipe: mdlm_di4c_lm1b}
owt: {status: supported, provenance: official}