project Refactor, CNN Classifier Basics

2020-02-19 21:11:42 +01:00
parent 8424251ca0
commit 78f0df8a2a
16 changed files with 622 additions and 560 deletions
@@ -1,10 +1,8 @@
-from lib.models.blocks import RecurrentModule, ConvModule, DeConvModule, Generator, LightningBaseModule
-from lib.models.losses import BinaryHomotopicLoss
+from lib.modules.blocks import LightningBaseModule
+from lib.modules.losses import BinaryHomotopicLoss
 from lib.objects.map import Map
 from lib.objects.trajectory import Trajectory

-import torch
-import torch.functional as F
 import torch.nn as nn

 nn.MSELoss