30 lines
644 B
Python
30 lines
644 B
Python
from lib.models.blocks import RecurrentModule, ConvModule, DeConvModule, Generator, LightningBaseModule
|
|
|
|
|
|
class CNNRouteGeneratorModel(LightningBaseModule):
|
|
|
|
@classmethod
|
|
def name(cls):
|
|
pass
|
|
|
|
def configure_optimizers(self):
|
|
pass
|
|
|
|
def validation_step(self, *args, **kwargs):
|
|
pass
|
|
|
|
def validation_end(self, outputs):
|
|
pass
|
|
|
|
def training_step(self, batch_xy, batch_nb, *args, **kwargs):
|
|
pass
|
|
|
|
def test_step(self, *args, **kwargs):
|
|
pass
|
|
|
|
def __init__(self, *params):
|
|
super(CNNRouteGeneratorModel, self).__init__(*params)
|
|
|
|
def forward(self, x):
|
|
pass
|