Update run.py

sunnweiwei · Feb 1, 2024 · a6afe33 · a6afe33
1 parent d3c1609
commit a6afe33
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/run.py b/run.py
@@ -1007,7 +1007,7 @@ def do_epoch_encode(model: Model, data, corpus, ids, tokenizer, batch_size, save
                                               shuffle=False, num_workers=16)
 
     collection, doc_code = our_encode(data_loader, model, 'doc')
-    doc_code = [0] * len(corpus)
+    # doc_code = [0] * len(corpus)
 
     print(collection.shape)
     index = build_index(collection, gpu=False)