Exemple #1
0
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)
Exemple #2
0
from catalyst.rl import registry

from .env import AtariEnvWrapper
from .critic import ConvActionCritic

registry.Environment(AtariEnvWrapper)
registry.Agent(ConvActionCritic)
Exemple #3
0
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)
Exemple #4
0
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)