34 lines
683 B
YAML
34 lines
683 B
YAML
defaults:
|
|
- diffusion_policy/config/train_diffusion_transformer_hybrid_workspace@_here_
|
|
- override /diffusion_policy/config/task@task: pusht_image
|
|
- _self_
|
|
|
|
exp_name: pusht_image_dit_imf_fullattn
|
|
|
|
policy:
|
|
_target_: diffusion_policy.policy.imf_transformer_hybrid_image_policy.IMFTransformerHybridImagePolicy
|
|
num_inference_steps: 1
|
|
n_head: 1
|
|
causal_attn: false
|
|
|
|
logging:
|
|
backend: swanlab
|
|
mode: online
|
|
name: ${exp_name}
|
|
resume: false
|
|
tags: ["${name}", "${task_name}", "${exp_name}", "swanlab"]
|
|
id: null
|
|
group: ${exp_name}
|
|
|
|
dataloader:
|
|
num_workers: 0
|
|
|
|
val_dataloader:
|
|
num_workers: 0
|
|
|
|
task:
|
|
env_runner:
|
|
n_envs: 1
|
|
n_test_vis: 0
|
|
n_train_vis: 0
|