65fe463
1
2
3
4
5
6
7
8
9
from abc import ABC, abstractmethod from torch.nn import Module class ModuleFactory(ABC): @abstractmethod def create(self) -> Module: pass