colab link updated

Author: awarebayes

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).

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))