SmartRick
SmartRick
from kiui.op import safe_normalize, dot ModuleNotFoundError: No module named 'kiui' Python 3.10.11 

 我正在基于秋叶训练器开发一款GUI训练管理工具,但发现可用api比较少,很多功能无法实现。例如获取任务的日志或者训练进度等信息,希望可以完善一下api能力。当前有一个硬性问题是无法获取任务对应的TensorBoard日志,保存的日志未和任务产生关联,因此通过查看源码发现在run_train方法中可以传入日志前缀,这样改动后即可在业务侧调用接口获取任务对应的日志数据,由此我获得到了日志的loss和训练step相关数据 ```python def run_train(toml_path: str, trainer_file: str = "./scripts/train_network.py", gpu_ids: Optional[list] = None, cpu_threads: Optional[int] = 2): log.info(f"Training started with config file / 训练开始,使用配置文件: {toml_path}") customize_env = os.environ.copy() customize_env["ACCELERATE_DISABLE_RICH"]...