feature(zc): add plan diffuser (#700)

opendilab · Oct 26, 2023 · 8834c21 · 8834c21
1 parent 88ce090
commit 8834c21
Show file tree

Hide file tree

Showing 22 changed files with 3,211 additions and 1 deletion.
diff --git a/ding/entry/serial_entry_offline.py b/ding/entry/serial_entry_offline.py
@@ -75,6 +75,8 @@ def serial_pipeline_offline(
     evaluator_env.seed(cfg.seed, dynamic_seed=False)
     set_pkg_seed(cfg.seed, use_cuda=cfg.policy.cuda)
     policy = create_policy(cfg.policy, model=model, enable_field=['learn', 'eval'])
+    if cfg.policy.collect.data_type == 'diffuser_traj':
+        policy.init_data_normalizer(dataset.normalizer)
 
     if hasattr(policy, 'set_statistic'):
         # useful for setting action bounds for ibc