Refactored vision backbone networks. Maniskill first pass.

Denys88 · Aug 31, 2024 · d073f4d · d073f4d
1 parent 3531fb8
commit d073f4d
Show file tree

Hide file tree

Showing 14 changed files with 457 additions and 635 deletions.
diff --git a/rl_games/algos_torch/model_builder.py b/rl_games/algos_torch/model_builder.py
@@ -2,7 +2,6 @@
 import rl_games.algos_torch
 from rl_games.algos_torch import network_builder, models
 
-
 NETWORK_REGISTRY = {}
 MODEL_REGISTRY = {}
 
@@ -20,12 +19,7 @@ def __init__(self):
         self.network_factory.register_builder('actor_critic', lambda **kwargs: network_builder.A2CBuilder())
         self.network_factory.register_builder('resnet_actor_critic',
                                             lambda **kwargs: network_builder.A2CResnetBuilder())
-        self.network_factory.register_builder('vision_actor_critic',
-                                            lambda **kwargs: network_builder.A2CVisionBuilder())
-        self.network_factory.register_builder('e2e_vision_actor_critic',
-                                            lambda **kwargs: network_builder.VisionBackboneBuilder())
 
-        self.network_factory.register_builder('rnd_curiosity', lambda **kwargs: network_builder.RNDCuriosityBuilder())
         self.network_factory.register_builder('soft_actor_critic', lambda **kwargs: network_builder.SACBuilder())
 
     def load(self, params):