last version of architecture

insight/insight_arch.py

@@ -1,3 +1,6 @@
+from torch import nn, optim
+import torch
+
 from torch import nn, optim
 import torch
 class Photoz_network(nn.Module):
@@ -41,14 +44,17 @@ class Photoz_network(nn.Module):
             nn.Linear(20, num_gauss)
         )
         
+        
     def forward(self, x):
         f = self.features(x)
         mu = self.measure_mu(f)
         sigma = self.measure_sigma(f)
         logmix_coeff = self.measure_coeffs(f)
-        
+                
         logmix_coeff = logmix_coeff - torch.logsumexp(logmix_coeff, 1)[:,None]
-
+        
         return mu, sigma, logmix_coeff
+
+