-
- Downloads
NCF changes to make it compatible with latest changes in master
* Pass the 'sigmoid' flag in NeuMF.forward as a bool tensor instead of a simple boolean. Required to make the model traceable (it?d be better to not have it an argument of forward at all, but keeping changes to a minimum) * Call prepare_model with dummy_input
Loading
Please register or sign in to comment