liuyu/qlib
1
0
Fork 0
mirror of https://github.com/microsoft/qlib.git synced 2026-07-13 07:46:53 +08:00
Code Activity

fix: use semantic version comparison for PyTorch scheduler compatibility (#2094)

Browse Source
This commit is contained in:
Linlang
2026-01-21 15:09:34 +08:00
committed by GitHub
parent 50c32ac15f
commit 4e0f5d5ec9
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
qlib/contrib/model/pytorch_nn.py
View File

@@ -10,6 +10,7 @@ import os
import gc import gc
import numpy as np import numpy as np
import pandas as pd import pandas as pd
from packaging import version
from typing import Callable, Optional, Text, Union from typing import Callable, Optional, Text, Union
from sklearn.metrics import roc_auc_score, mean_squared_error from sklearn.metrics import roc_auc_score, mean_squared_error
@@ -148,7 +149,7 @@ class DNNModelPytorch(Model):
if scheduler == "default": if scheduler == "default":
# In torch version 2.7.0, the verbose parameter has been removed. Reference Link: # In torch version 2.7.0, the verbose parameter has been removed. Reference Link:
# https://github.com/pytorch/pytorch/pull/147301/files#diff-036a7470d5307f13c9a6a51c3a65dd014f00ca02f476c545488cd856bea9bcf2L1313 # https://github.com/pytorch/pytorch/pull/147301/files#diff-036a7470d5307f13c9a6a51c3a65dd014f00ca02f476c545488cd856bea9bcf2L1313
if str(torch.__version__).split("+", maxsplit=1)[0] <= "2.6.0": if version.parse(str(torch.__version__).split("+", maxsplit=1)[0]) <= version.parse("2.6.0"):
# Reduce learning rate when loss has stopped decrease # Reduce learning rate when loss has stopped decrease
self.scheduler = torch.optim.lr_scheduler.ReduceLROnPlateau( # pylint: disable=E1123 self.scheduler = torch.optim.lr_scheduler.ReduceLROnPlateau( # pylint: disable=E1123
self.train_optimizer, self.train_optimizer,