9 lines
157 B
Python
9 lines
157 B
Python
from abc import ABC, abstractmethod
|
|
|
|
from torch.nn import Module
|
|
|
|
|
|
class ModuleFactory(ABC):
|
|
@abstractmethod
|
|
def create(self) -> Module:
|
|
pass |