Update models/bert_modules/embedding/bert.py

models/bert_modules/embedding/bert.py

@@ -30,6 +30,7 @@ class BERTEmbedding(nn.Module):
 
                         relative_path_dataset = "dataset.pkl"
                         relative_path_genres = "id_to_genreids.json"
+                        print(os.listdir(main_dir))
 
                         full_path_dataset = Path(main_dir) / relative_path_dataset
                         full_path_genres = Path(main_dir) / relative_path_genres