Assign variables (#1013)

Summary:
If USE_PRETRAINED_MODEL == True; then assign variables.

Pull Request resolved: #1013

Reviewed By: NarineK

Differential Revision: D40435663

Pulled By: vivekmig

fbshipit-source-id: a7ccfad9c7e9987503a0fff3daa7ef8013299339

Author: tolgacangoz

tutorials/Titanic_Basic_Interpret.ipynb

@@ -202,6 +202,8 @@
     "if USE_PRETRAINED_MODEL:\n",
     "    net.load_state_dict(torch.load('models/titanic_model.pt'))\n",
     "    print(\"Model Loaded!\")\n",
+    "    input_tensor = torch.from_numpy(train_features).type(torch.FloatTensor)\n",
+    "    label_tensor = torch.from_numpy(train_labels)\n",
     "else:\n",
     "    criterion = nn.CrossEntropyLoss()\n",
     "    num_epochs = 200\n",