wangzizhao / CausalDynamicsLearning

26 stars 9 forks source link

what is the difference between manipulation_dynamic_learning.json and manipulaion_task_learning.json #3

Open 945716994 opened 1 year ago

945716994 commented 1 year ago

Did in "manipulation_dynamic_learning" just learn a transition model? And not specifiy to a task? When I change the "env_name" to "Causal" in polcy_params.json just like in "manipulation_dynamic_learning".json, and run the main_policy.py , I got a some error: "AttributeError: 'Causal' object has no attribute 'shape'"

Can you explain the two file's mean?(xxxx_dynamic_xx.json, xxx_task_xxx.json) Thanks!

wangzizhao commented 1 year ago

Yes, as you guess, xxxx_dynamic_xx.json only learns a dynamics model that's not specific to a task. Then xxx_task_xxx.json load and freeze the dynamics model (the load path is specified by training_params.load_inference) and learns to solve a specific task.