New Dataset Generator, How to differentiate the loss function?

2020-02-18 21:58:31 +01:00
parent 61c5cb44a0
commit 8424251ca0
13 changed files with 250 additions and 39 deletions
--- a/lib/models/cnn.py
+++ b/lib/models/cnn.py
@ -3,9 +3,7 @@ from lib.models.blocks import RecurrentModule, ConvModule, DeConvModule, Generat

 class CNNRouteGeneratorModel(LightningBaseModule):

-    @classmethod
-    def name(cls):
-        pass
+    name = 'CNNRouteGenerator'

    def configure_optimizers(self):
        pass