Remove self.device to prevent import requirement of torch for prediction (#617) (5de0a8e9) · Commits · Arthur Batel / cornac · GitLab

Skip to content

GitLab

Explore

Sign in

Primary navigation

Project

C

cornac
- Activity
- Members
- Labels
- Issues
  0
- Issue boards
- Milestones
- Wiki
- Environments
- Terraform modules
- Incidents

Snippets Groups Projects

5de0a8e9

Unverified Commit 5de0a8e9 authored 1 year ago by darrylong Committed by GitHub 1 year ago

Downloads
- Patches
- Plain Diff

Remove self.device to prevent import requirement of torch for prediction (#617)

parent ae7ba860

No related branches found

No related tags found

No related merge requests found

Changes 2

Hide whitespace changes

Inline Side-by-side

Showing

cornac/models/lightgcn/lightgcn.py 1 addition, 2 deletions

cornac/models/lightgcn/lightgcn.py
cornac/models/lightgcn/recom_lightgcn.py 3 additions, 3 deletions

cornac/models/lightgcn/recom_lightgcn.py

with 4 additions and 5 deletions

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 sign in to comment