From 629b91581e2b835ac1b5fdb652765518e8b2a0e5 Mon Sep 17 00:00:00 2001 From: zhengbw0324 <18735382001@163.com> Date: Tue, 4 Oct 2022 13:14:01 +0800 Subject: [PATCH] FIX: fix fwfm bug --- recbole/model/context_aware_recommender/fwfm.py | 1 - 1 file changed, 1 deletion(-) diff --git a/recbole/model/context_aware_recommender/fwfm.py b/recbole/model/context_aware_recommender/fwfm.py index a5f7ab2cd..7013c7848 100644 --- a/recbole/model/context_aware_recommender/fwfm.py +++ b/recbole/model/context_aware_recommender/fwfm.py @@ -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