32 lines
670 B
Python
32 lines
670 B
Python
from lib.modules.utils import LightningBaseModule
|
|
from lib.modules.blocks import ConvModule
|
|
|
|
class ConvHomDetector(LightningBaseModule):
|
|
|
|
name = 'CNNHomotopyClassifier'
|
|
|
|
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(ConvHomDetector, self).__init__(*params)
|
|
|
|
self.conv1 = ConvModule(self.dataset.map_shape
|
|
|
|
)
|
|
|
|
|
|
def forward(self, x):
|
|
pass |