From ae720aa470d199a80f5691f1c8b6695601993cc1 Mon Sep 17 00:00:00 2001
From: zbw <18735382001@163.com>
Date: Tue, 19 Mar 2024 10:08:05 +0800
Subject: [PATCH] Fix: fix the loss coefficient of s3rec.

---
 recbole/properties/model/S3Rec.yaml | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/recbole/properties/model/S3Rec.yaml b/recbole/properties/model/S3Rec.yaml
index 6d84c5b2c..34e877ebf 100644
--- a/recbole/properties/model/S3Rec.yaml
+++ b/recbole/properties/model/S3Rec.yaml
@@ -9,12 +9,12 @@ layer_norm_eps: 1e-12           # (float) A value added to the denominator for n
 initializer_range: 0.02         # (float) The standard deviation for normal initialization.
 item_attribute: 'class'         # (str) The item features used as attributes for pre-training.
 mask_ratio: 0.2                 # (float) The probability for a item replaced by MASK token.
-aap_weight: 1.0                 # (float) The weight for Associated Attribute Prediction loss.
-mip_weight: 0.2                 # (float) The weight for Masked Item Prediction loss.
+aap_weight: 0.2                 # (float) The weight for Associated Attribute Prediction loss.
+mip_weight: 1.0                 # (float) The weight for Masked Item Prediction loss.
 map_weight: 1.0                 # (float) The weight for Masked Attribute Prediction loss.
 sp_weight: 0.5                  # (float) The weight for Segment Prediction loss.
 train_stage: 'pretrain'         # (str) The training stage. Range in ['pretrain', 'finetune'].
 pretrain_epochs: 500            # (int) The epochs of pre-training.
-save_step: 10                   # (int) Save pre-trained model every pre-training epochs.
+save_step: 50                   # (int) Save pre-trained model every pre-training epochs.
 pre_model_path: ''              # (str) The path of pretrained model.
 loss_type: 'CE'                 # (str) The type of loss function. Range in ['BPR', 'CE'].
\ No newline at end of file