File size: 157 Bytes
6a62ffb
 
 
 
 
 
 
 
 
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