\n",
+ " \n",
+ " 0 | \n",
+ " embed | \n",
+ " torch.nn.modules.sparse.Embedding | \n",
+ " False | \n",
+ " 0.853265 | \n",
+ "
\n",
+ " \n",
+ " 1 | \n",
+ " project | \n",
+ " torch.nn.modules.linear.Linear | \n",
+ " False | \n",
+ " 0.494231 | \n",
+ "
\n",
+ " \n",
+ " 2 | \n",
+ " unembed | \n",
+ " torch.nn.modules.linear.Linear | \n",
+ " False | \n",
+ " 0.581503 | \n",
+ "
\n",
+ " \n",
+ " 3 | \n",
+ " | \n",
+ " __main__.Model | \n",
+ " False | \n",
+ " NaN | \n",
+ "
\n",
+ " \n",
+ " 4 | \n",
+ " | \n",
+ " __main__.Model | \n",
+ " True | \n",
+ " NaN | \n",
+ "
\n",
+ " \n",
+ " 5 | \n",
+ " unembed | \n",
+ " torch.nn.modules.linear.Linear | \n",
+ " True | \n",
+ " 0.105266 | \n",
+ "
\n",
+ " \n",
+ " 6 | \n",
+ " project | \n",
+ " torch.nn.modules.linear.Linear | \n",
+ " True | \n",
+ " 0.112392 | \n",
+ "
\n",
+ " \n",
+ " 7 | \n",
+ " embed | \n",
+ " torch.nn.modules.sparse.Embedding | \n",
+ " True | \n",
+ " 0.068816 | \n",
+ "
\n",
+ " \n",
+ "