You are viewing a plain text version of this content. The canonical link for it is here.
Posted to dev@bluemarlin.apache.org by GitBox <gi...@apache.org> on 2022/02/21 14:49:04 UTC
[GitHub] [incubator-bluemarlin] jimmylao commented on issue #51: [BLUEMARLIN-28] : For DIN-Lookalike model, train.py runs only if line 40 in model.py is commented.
jimmylao commented on issue #51:
URL: https://github.com/apache/incubator-bluemarlin/issues/51#issuecomment-1046956459
@Bimlesh759-AI
It's ok to comment out line 40 since it's not currently used in the model.
The variable "user_emb_w" is reserved in the model in case to use more user specific features. It is claimed to have the shape of user_count x hidden_units (128). In your case, I guess "user_count" (number of users) is huge and triggers a memory overflow error.
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: dev-unsubscribe@bluemarlin.apache.org
For queries about this service, please contact Infrastructure at:
users@infra.apache.org