Update loss criteria constructor (#1711)
This commit is contained in:
1 parent
799724108f
commit
8bc0027afc
4 files changed
+18
-15
No files matched your search
Vendored
+1
-1
@@ -1199,7 +1199,7 @@
|
||||
"\n",
|
||||
"m1 = lambda x: x * torch.sigmoid(x)\n",
|
||||
"m2 = torch.nn.SiLU()\n",
|
||||
"profile(x=torch.randn(16, 3, 640, 640), [m1, m2], n=100)"
|
||||
"profile(x=torch.randn(16, 3, 640, 640), ops=[m1, m2], n=100)"
|
||||
],
|
||||
"execution_count": null,
|
||||
"outputs": []
|
||||
|
||||
Reference in new issue
Block a user