Update loss criteria constructor (#1711)

This commit is contained in:
Glenn Jocher authored and GitHub committed 2020-12-16 08:39:35 -08:00
1 parent 799724108f
commit 8bc0027afc
4 files changed
+18 -15

No files matched your search

+1 -1
View File
@@ -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": []