文件名称:contrastive-forward-model
文件大小:46KB
文件格式:ZIP
更新时间:2024-05-24 13:12:05
Python
对比正向模型 这是用于重现实验的代码。 安装 该项目是使用Python 3.7.6运行的。 所有依赖项都在requirements.txt文件中,我们建议创建一个虚拟环境,然后通过pip install -r requirements.txt 。 您还需要安装具有相关的绳索和布料环境的自定义和自定义。 您必须在自定义dm_control库中使用cfm分支。 请注意, dm_control需要使用Mujoco模拟器。 最后,您将需要安装此回购为PIP封装: cd contrastive-forward-model; pip install -e . cd contrastive-forward-model; pip install -e . 跑步 收集和运行数据的步骤如下。 您可以使用-h标志显示更多可定制的选项。 通过运行python sample_trajectories.py收集数
【文件预览】:
contrastive-forward-model-master
----sample_trajectories.py(4KB)
----cfm()
--------models.py(11KB)
--------evaluate_planning.py(14KB)
--------utils.py(2KB)
--------dataset.py(7KB)
--------visualize()
--------train_cfm.py(7KB)
--------__init__.py(0B)
--------baselines()
--------env()
----requirements.txt(171B)
----out()
--------.gitkeep(0B)
----setup.py(220B)
----.gitignore(38B)
----run_evaluation.py(2KB)
----README.md(2KB)
----run_baselines()
--------run_train_visual_forward.py(2KB)
--------run_train_planet.py(2KB)
--------run_train_joint_dynamics.py(2KB)
--------run_train_autoencoder.py(2KB)
----process_dataset.py(4KB)
----run_train.py(2KB)