search

ShomyLiu / Neu-Review-Rec

A Toolkit for Neural Review-based Recommendation models with Pytorch.
http://shomy.top/2019/12/31/neu-review-rec/
168 stars 54 forks source link

DAML模型 #3

Open jerryLee80 opened 4 years ago

jerryLee80 commented 4 years ago

请问DAML模型实现了吗？我在运行的时候，第一次循环就提示GPU内存溢出（错误应该发生在loss.backward()处，但我未找出原因），而其他2个模型都是可以正常运行的。

ShomyLiu commented 4 years ago

你好！ DAML是实现了的。 但是由于DAML是完全的Context-aware，需要维护一个大矩阵，因此占用显存会大一些。建议改小batch_size

jerryLee80 commented 4 years ago

谢谢。我看默认是64，以前改过32，还是报错，所以没再改小了。我用的是1080Ti。另外，能再将num_fea取值为1，2，3的含义解释下吗？非常感谢。

ShomyLiu commented 4 years ago

DAML可以试试小数据集，比如Musical,Office等。 一般feature有三类：

相关说明会更新到readme中。

jerryLee80 commented 4 years ago

感谢！batch_size改为24，基本就可以了。 期待新模型的加入及详细说明的更新。

luoxintao commented 4 years ago

谢谢。我看默认是64，以前改过32，还是报错，所以没再改小了。我用的是1080Ti。另外，能再将num_fea取值为1，2，3的含义解释下吗？非常感谢。

16G的显卡，batchsize改为32才跑起来。。

yinzhiqiangluvlzx commented 3 years ago

楼主，方便请教一个问题嘛。python3 main.py test --pth_path="./checkpoints/THE_PTH_PATH" --model=DeepCoNN 这个命令是需要把THE_PTH_PATH替换成保存模型的名字是吧。但是pth_path = "" # the saved pth path for test 这个写成了保存模型的路径依然报错，你有遇到过这种情况么？或者你可以跟我说下你那个测试集是怎么启动的么。十分感谢啦

FKCHAN commented 3 years ago

你好，因为跑不起来，所以用了三块显卡并行，但是保存模型的时候报错了， AttributeError: 'DataParallel' object has no attribute 'save'

ShomyLiu commented 3 years ago

@FKCHAN 你好， DataParallel 包装的模型 不能直接使用save 来保存模型, 参考：https://pytorch.org/tutorials/beginner/saving_loading_models.html#saving-torch-nn-dataparallel-models