Source code for catalyst.rl.core.agent
from abc import ABC, abstractmethod
import torch.nn as nn
[docs]class ActorSpec(ABC, nn.Module):
@property
@abstractmethod
def policy_type(self) -> str:
pass
[docs] @abstractmethod
def forward(self, state, logprob=None, deterministic=False):
pass
[docs]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
[docs] @abstractmethod
def forward(self, *args, **kwargs):
pass