PLM4NewsRec
所属分类：人工智能/神经网络/深度学习
开发工具：Python
文件大小：17KB
下载次数：0
上传日期：2022-05-12 03:40:02
上 传 者：
sh-1993
说明： PLM4新闻报道，，
(PLM4NewsRec,,)
文件列表:
dataloader.py (12277, 2022-05-12)
demo.sh (2402, 2022-05-12)
initenvGPU.sh (575, 2022-05-12)
metrics.py (792, 2022-05-12)
model_bert.py (14497, 2022-05-12)
parameters.py (5169, 2022-05-12)
preprocess.py (11998, 2022-05-12)
run.py (13376, 2022-05-12)
run.sh (91, 2022-05-12)
streaming.py (6595, 2022-05-12)
utils.py (3728, 2022-05-12)
# PLM-NR
This pipeline is based on the horovod framework. We run experiments on 4 GPUs. You need to split the original behaviors.tsv file in the MIND dataset into X parts (named as behaviors_x.tsv, X {0, 1, 2, ..., X}) if you use X GPUs to train.
Run the run.sh script to train and test.
近期下载者：
相关文件：
收藏者：