search

RUCDM / KB4Rec

This is the data for KB4Rec
328 stars 52 forks source link

AssertionError: Checkpoint file not found! #10

Closed Zhang-Each closed 2 years ago

Zhang-Each commented 2 years ago

Found error when runing UPGAN code, it seems that the pre-trained embeddings are not released. How to solve such problem.

Zhang-Each commented 2 years ago

When I Run CUDA_VISIBLE_DEVICES=2 python main_pretrain.py command, the following file is not found.

checkpoint/good_pretrain/book/init/book-DistMult-bce-decay.ckpt

RichardHGL commented 2 years ago

Please refer to https://github.com/RichardHGL/UPGAN