iicd/autodl-projects
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix bugs in update_gpu in procedures

Browse Source
This commit is contained in:
D-X-Y
2021-03-06 19:27:05 -08:00
parent 55c9734c31
commit 53e1441c8d
2 changed files with 21 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
lib/procedures/q_exps.py
View File

@@ -12,10 +12,18 @@ from qlib.log import get_module_logger
def update_gpu(config, gpu):
config = config.copy()
if "task" in config and "moodel" in config["task"] and "GPU" in config["task"]["model"]:
config["task"]["model"]["GPU"] = gpu
elif "model" in config and "GPU" in config["model"]:
config["model"]["GPU"] = gpu
if "task" in config and "model" in config["task"]:
if "GPU" in config["task"]["model"]:
config["task"]["model"]["GPU"] = gpu
elif "kwargs" in config["task"]["model"] and "GPU" in config["task"]["model"]["kwargs"]:
config["task"]["model"]["kwargs"]["GPU"] = gpu
elif "model" in config:
if "GPU" in config["model"]:
config["model"]["GPU"] = gpu
elif "kwargs" in config["model"] and "GPU" in config["model"]["kwargs"]:
config["model"]["kwargs"]["GPU"] = gpu
elif "kwargs" in config and "GPU" in config["kwargs"]:
config["kwargs"]["GPU"] = gpu
elif "GPU" in config:
config["GPU"] = gpu
return config