Upload RecBot

config.json

@@ -1,4 +1,5 @@
 {
+  "LOAD_SAVE_IGNORES": [],
   "_commit_hash": null,
   "aggregator": null,
   "architectures": [
@@ -6,31 +7,45 @@
   ],
   "datahandler": null,
   "generator": {
+    "LOAD_SAVE_IGNORES": [],
     "_commit_hash": null,
+    "bos_token_id": 1,
     "context_size": 256,
-    "embedding_path": "embeddings/glove.840B.300d.h5",
+    "eos_token_id": 2,
     "hidden_size": 256,
     "moduleName": "SwitchingDecoder",
     "moduleType": "generator",
     "num_layers": 1,
+    "output_map": null,
     "peephole": false,
+    "task_embedding.dtype": "torch.float32",
+    "task_embedding.shape": [
+      15005,
+      300
+    ],
     "transformers_version": null
   },
   "manager": {
+    "LOAD_SAVE_IGNORES": [],
     "_commit_hash": null,
     "conversation_encoder_hidden_size": 256,
     "moduleName": "RedialManager",
     "moduleType": "manager",
+    "n_movies": 6924,
+    "num_vocab": 15005,
+    "output_map": null,
     "transformers_version": null
   },
   "processor": [
     {
+      "LOAD_SAVE_IGNORES": [],
       "_commit_hash": null,
       "conv_bidrectional": false,
       "conversation_encoder_hidden_size": 256,
       "conversation_encoder_num_layers": 1,
       "moduleName": "HRNN",
       "moduleType": "processor",
+      "output_map": null,
       "sentence_encoder_hidden_size": 256,
       "sentence_encoder_model": "princeton-nlp/unsup-simcse-roberta-base",
       "sentence_encoder_num_layers": 1,
@@ -39,6 +54,7 @@
       "use_movie_occurrences": false
     },
     {
+      "LOAD_SAVE_IGNORES": [],
       "_commit_hash": null,
       "_name_or_path": "ckpt/sa/model_best",
       "hrnn_params": {
@@ -62,11 +78,13 @@
         "r_seen": 3,
         "r_suggested": 1
       },
+      "output_map": null,
       "return_liked_probability": true,
       "transformers_version": null
     }
   ],
   "recommender": {
+    "LOAD_SAVE_IGNORES": [],
     "_commit_hash": null,
     "_name_or_path": "ckpt/autorec/db_model_best",
     "architectures": [
@@ -80,6 +98,7 @@
     "moduleName": "AutoRec",
     "moduleType": "recommender",
     "n_movies": 6924,
+    "output_map": null,
     "torch_dtype": "float32",
     "transformers_version": null
   },

pytorch_model.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:5439cde26480d1d634e75b002d9a212c1490ed540cf88d656e7725da96a1cf14
-size 1147029377
+oid sha256:67cb3783d96dbc98ba12a66b972a77e150192025f50d0bb2521aa811e83c2d4c
+size 149708927