from catalyst.rl import registry from src.env import AnimalEnvWrapper from src.actor import AnimalActor from src.critic import AnimalStateCritic registry.Environment(AnimalEnvWrapper) registry.Agent(AnimalActor) registry.Agent(AnimalStateCritic)
from catalyst.rl import registry from .env import AtariEnvWrapper from .critic import ConvActionCritic registry.Environment(AtariEnvWrapper) registry.Agent(ConvActionCritic)
from catalyst.rl import registry from .env import MiniGridEnvWrapper from .actor import ConvActor from .critic import ConvCritic, ConvQCritic registry.Environment(MiniGridEnvWrapper) registry.Agent(ConvActor) registry.Agent(ConvCritic) registry.Agent(ConvQCritic)
from catalyst.rl import registry from src.env import SkeletonEnvWrapper from src.actor import SkeletonActor from src.critic import SkeletonStateCritic, SkeletonStateActionCritic registry.Environment(SkeletonEnvWrapper) registry.Agent(SkeletonActor) registry.Agent(SkeletonStateCritic) registry.Agent(SkeletonStateActionCritic)