module updates

This commit is contained in:
Glenn Jocher committed 2020-06-08 22:13:01 -07:00
1 parent d187459dd4
commit ff02ae0869
3 files changed
+16 -43

No files matched your search

+1 -3
View File
@@ -176,9 +176,7 @@ def parse_model(md, ch): # model_dict, input_channels(3)
elif m is nn.BatchNorm2d:
args = [ch[f]]
elif m is Concat:
c2 = sum([ch[x] for x in f])
elif m is Origami:
c2 = ch[f] * 5
c2 = sum([ch[-1 if x == -1 else x + 1] for x in f])
elif m is Detect:
f = f or list(reversed([(-1 if j == i else j - 1) for j, x in enumerate(ch) if x == no]))
else: