Source code for catalyst.rl.core.agent

from abc import ABC, abstractmethod

import torch.nn as nn


class ActorSpec(ABC, nn.Module):
    @property
    @abstractmethod
    def policy_type(self) -> str:
        pass

    @abstractmethod
    def forward(self, state, logprob=None, deterministic=False):
        pass


class CriticSpec(ABC, nn.Module):
    @property
    @abstractmethod
    def num_outputs(self) -> int:
        pass

    @property
    @abstractmethod
    def num_atoms(self) -> int:
        pass

    @property
    @abstractmethod
    def distribution(self) -> str:
        pass

    @property
    @abstractmethod
    def values_range(self) -> tuple:
        pass

    @abstractmethod
    def forward(self, *args, **kwargs):
        pass