Skip to content

Commit

Permalink
FIX: fix fwfm bug
Browse files Browse the repository at this point in the history
  • Loading branch information
zhengbw0324 committed Oct 4, 2022
1 parent 1f50ee3 commit 629b915
Showing 1 changed file with 0 additions and 1 deletion.
1 change: 0 additions & 1 deletion recbole/model/context_aware_recommender/fwfm.py
Original file line number Diff line number Diff line change
Expand Up @@ -61,7 +61,6 @@ def __init__(self, config, dataset):
self.weight = torch.randn(
self.num_fields, self.num_fields, 1, requires_grad=True, device=self.device
)

self.loss = nn.BCEWithLogitsLoss()

# parameters initialization
Expand Down

0 comments on commit 629b915

Please sign in to comment.