Skip to content
Snippets Groups Projects
Commit 81787436 authored by Guy Jacob's avatar Guy Jacob
Browse files

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
parent c4d020d9
No related branches found
No related tags found
No related merge requests found
Loading
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment