Commit 49942609 authored by connor.hainje@pnnl.gov's avatar connor.hainje@pnnl.gov
Browse files

Formatting

parent 4132573d
...@@ -162,8 +162,9 @@ class SimpleNet(GeneralNet): ...@@ -162,8 +162,9 @@ class SimpleNet(GeneralNet):
'probabilities' that each particle was of each possible particle 'probabilities' that each particle was of each possible particle
type. type.
""" """
n = self.n_detector
outs = [ outs = [
self.fcs[i](x[:, i * self.n_detector : (i + 1) * self.n_detector]) self.fcs[i](x[:, i * n : (i + 1) * n])
for i in range(self.n_class) for i in range(self.n_class)
] ]
out = torch.cat(outs, dim=1) out = torch.cat(outs, dim=1)
......
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment