[💡SUG] 能否提供SGL实验的详细超参数

[zzyo-yu]

你好，我希望在你们工作的基础上进行研究，但我需要你们在movie-len上进行的实验的详细参数文件，不知是否可以提供

[hyp1231]

你好，我们在 General - ML-1M 里列出了全部的调参范围和最优参数。如果你希望复现最优结果，可以直接将此页面的 properties 和最优参数拼成一个新的 .yaml，以 SGL 举例：

# dataset config
field_separator: "\t"
seq_separator: " "
USER_ID_FIELD: user_id
ITEM_ID_FIELD: item_id
RATING_FIELD: rating
NEG_PREFIX: neg_
LABEL_FIELD: label
load_col:
    inter: [user_id, item_id, rating]
val_interval:
    rating: "[3,inf)"
unused_col: 
    inter: [rating]

# training and evaluation
epochs: 500
train_batch_size: 4096
valid_metric: MRR@10
eval_batch_size: 4096000

# best Hyper-parameter for SGL
learning_rate: 0.002
n_layers: 3
reg_weight: 0.0001
ssl_tau: 0.5
drop_ratio: 0.1
ssl_weight: 0.005

假设此文件名为 sgl_best.yaml，则可以将其放在根目录，并输下述命令运行

python run_recbole_gnn.py -m SGL -d ml-1m --config_files=sgl_best.yaml

[zzyo-yu]

非常感谢！我还有一个疑问，我在Amazon Book上的实验提示了显存溢出的错误（使用2060显卡，6G显存），您知道这个问题应该如何解决吗？

[hyp1231]

Book 数据集上好像确实是需要较大的显存的。