21 lines
472 B
Python
21 lines
472 B
Python
|
from torch import nn
|
||
|
|
from configs.config import ConfigManager
|
||
|
|
|
||
|
|
|
||
|
|
class Pipeline(nn.Module):
|
||
|
|
TRAIN_MODE: str = "train"
|
||
|
|
TEST_MODE: str = "test"
|
||
|
|
|
||
|
|
def __init__(self, pipeline_config):
|
||
|
|
super(Pipeline, self).__init__()
|
||
|
|
|
||
|
|
self.modules_config = ConfigManager.get("modules")
|
||
|
|
self.device = ConfigManager.get("settings", "general", "device")
|
||
|
|
|
||
|
|
def forward(self, data, mode):
|
||
|
|
pass
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
pass
|