v2

2021-02-26 16:12:51 +00:00
parent c895924c99
commit b74255e1f3
74 changed files with 11326 additions and 537 deletions
--- a/models/cell_infers/tiny_network.py
+++ b/models/cell_infers/tiny_network.py
@@ -55,4 +55,4 @@ class TinyNetwork(nn.Module):
    out = out.view(out.size(0), -1)
    logits = self.classifier(out)

-    return out, logits
+    return logits, out