Skip to content

Commit

Permalink
Correcting paths in rec system tutorial
Browse files Browse the repository at this point in the history
Summary: Correcting paths to open source version.

Reviewed By: Yonathae

Differential Revision: D56471234

fbshipit-source-id: 506f9058ae408c8aeea23d6f2ad38a9031af703d
  • Loading branch information
rodrigodesalvobraz authored and facebook-github-bot committed Apr 23, 2024
1 parent 4175432 commit 057a48a
Showing 1 changed file with 2 additions and 2 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -281,8 +281,8 @@
"device_id = 0 if torch.cuda.is_available() else -1\n",
"\n",
"model = SequenceClassificationModel(100).to(device)\n",
"model.load_state_dict(torch.load(\"/data/sandcastle/boxes/fbsource/fbcode/pearl/tutorials/single_item_recommender_system_example/env_model_state_dict.pt\"))\n",
"actions = torch.load(\"/data/sandcastle/boxes/fbsource/fbcode/pearl/tutorials/single_item_recommender_system_example/news_embedding_small.pt\")\n",
"model.load_state_dict(torch.load(\"Pearl/tutorials/single_item_recommender_system_example/env_model_state_dict.pt\"))\n",
"actions = torch.load(\"Pearl/tutorials/single_item_recommender_system_example/news_embedding_small.pt\")\n",
"history_length = 8\n",
"env = RecEnv(list(actions.values())[:100], model, history_length)\n",
"observation, action_space = env.reset()\n",
Expand Down

0 comments on commit 057a48a

Please sign in to comment.