colab link updated

awarebayes · Oct 28, 2019 · 84c47d9 · 84c47d9
1 parent 640a225
commit 84c47d9
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/readme.md b/readme.md
@@ -90,7 +90,7 @@ This is my school project. It focuses on Reinforcement Learning for personalized
 ## Getting Started
 
 <p align="center"> 
-<a href="https://colab.research.google.com/drive/1qGjgLYuTpJTiW_P1pbs0zEgRa9BhZckw"><img src="./res/get_started.png"></a>
+<a href="https://colab.research.google.com/drive/1xWX4JQvlcx3mizwL4gB0THEyxw6LsXTL"><img src="./res/get_started.png"></a>
 </p>
 
 p.s. Image is clickable. here is [direct link](https://colab.research.google.com/drive/1qGjgLYuTpJTiW_P1pbs0zEgRa9BhZckw).

diff --git a/recnn/data/utils.py b/recnn/data/utils.py
@@ -130,7 +130,7 @@ def prepare_batch_static_size(batch, item_embeddings_tensor=False, frame_size=10
 
 def make_items_tensor(items_embeddings_key_dict, include_zero=True):
     if include_zero:
-        items_embeddings_key_dict[0] = torch.zeros(128)
+        items_embeddings_key_dict[0] = torch.zeros([128, ])
     keys = list(sorted(items_embeddings_key_dict.keys()))
     key_to_id = dict(zip(keys, range(len(keys))))
     id_to_key = dict(zip(range(len(keys)), keys))