静宝 2023/5/11 21:41:33 if args.version is None and (v := args.conf.find("version_")) != -1: args.version = int(args.conf[v + 8:args.conf.find("/config")]) print(f"[INFO] Loaded version {args.version} from config file") if args.version is not None: logger = loggers.TensorBoardLogger(save_dir=args.exps_folder, name=expname, version=args.version) else: logger = loggers.TensorBoardLogger(save_dir=args.exps_folder, name=expname) 静宝 2023/5/11 21:41:49 if args.device_ids is None: args.device_ids = GPUtil.getAvailable(order='memory', limit=1, maxLoad=0.5, maxMemory=0.5, includeNan=False, excludeID=[], excludeUUID=[]) print("Selected GPU {} automatically".format(args.device_ids[0]))
时间: 2023-06-16 14:08:16 浏览: 400
这段代码看起来像是一个 Python 脚本中的一部分,其作用是根据命令行参数和配置文件来确定 tensorboard 日志的保存路径和版本号,并选择一个可用的 GPU 设备。其中,如果配置文件中存在 version_ 字段,则会从中解析出版本号并打印出来;如果没有指定 device_ids,则会根据 GPU 的内存和负载情况自动选择一个可用的设备。
阅读全文