Models *.yaml reformat (#3875)
This commit is contained in:
1 parent
bd88e7f4f2
commit
81b31824f5
18 files changed
+184
-218
No files matched your search
+1
-1
@@ -154,7 +154,7 @@ class Model(nn.Module):
|
||||
|
||||
x = m(x) # run
|
||||
y.append(x if m.i in self.save else None) # save output
|
||||
|
||||
|
||||
if feature_vis and m.type == 'models.common.SPP':
|
||||
feature_visualization(x, m.type, m.i)
|
||||
|
||||
|
||||
Reference in new issue
Block a user