🚀 [RofuncRL] Add pre-trained model FrankaCabinetRofuncRLPPO

Skylark0924 · Oct 13, 2023 · dfeb15a · dfeb15a
1 parent 9bd8ac2
commit dfeb15a
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/rofunc/config/learning/model_zoo.json b/rofunc/config/learning/model_zoo.json
@@ -15,7 +15,7 @@
     "md5": "6ec13ea27b8be8500d91208fdee3ead2"
   },
   "FrankaCabinetRofuncRLPPO.pth": {
-    "url": "https://drive.google.com/uc?id=&export=download",
+    "url": "https://drive.google.com/uc?id=13h4Sp2aO_1UyiSoHojYMUsVHHglKcmii&export=download",
     "name": "FrankaCabinetRofuncRLPPO.pth",
     "md5": ""
   },

diff --git a/rofunc/learning/RofuncRL/tasks/__init__.py b/rofunc/learning/RofuncRL/tasks/__init__.py
@@ -14,7 +14,7 @@
 from .ase.humanoid_reach import HumanoidReachTask
 from .ase.humanoid_strike import HumanoidStrikeTask
 from .ase.humanoid_view_motion import HumanoidViewMotionTask
-from .omniisaacgym.ant import AntOmniTask
+# from .omniisaacgym.ant import AntOmniTask
 
 
 task_map = {
@@ -36,5 +36,5 @@
     "HumanoidASEStrikeSwordShield": HumanoidStrikeTask,
     "HumanoidViewMotion": HumanoidViewMotionTask,
     # OmniIsaacGym tasks
-    "AntOmni": AntOmniTask,
+    # "AntOmni": AntOmniTask,
 }