Trainer 和 wandb
时间: 2023-10-19 12:26:11 浏览: 241
wandb:一款轻量级的机器学习可视化工具 该项目是用于可视化和跟踪机器学习实验的工具,通过几行代码就可以实现跟踪、比较和可视化
Trainer和wandb_logger是PyTorch Lightning框架中用于训练和记录实验结果的两个关键组件。Trainer是PyTorch Lightning中的训练器,它负责整个训练过程的管理和控制。而wandb_logger是WandbLogger接口,它是用于将训练过程中的各类数据记录到Wandb平台的工具。
Trainer对象可以通过设置不同的超参数来控制训练的过程和行为。比如可以手动定义超参数,或者使用默认设置,每次运行都会自动记录到Wandb平台。
WandbLogger提供了一系列方法来记录各类数据到Wandb平台。在使用时,可以先初始化一个WandbLogger对象,然后将其作为logger传入Trainer中,就可以开始记录实验结果了。可以使用WandbLogger.watch()方法来设置需要记录的数据,比如梯度、模型拓扑等。此外,还可以设置日志的频率、是否记录图形等选项。 <span class="em">1</span><span class="em">2</span><span class="em">3</span>
#### 引用[.reference_title]
- *1* [lightning-kitti:使用 Pytorch-Lightning 进行语义分割](https://download.csdn.net/download/weixin_42131728/19226396)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v93^chatsearchT3_2"}}] [.reference_item style="max-width: 50%"]
- *2* *3* [wandb在pytorch lightning中的使用](https://blog.csdn.net/zzfive/article/details/125716913)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v93^chatsearchT3_2"}}] [.reference_item style="max-width: 50%"]
[ .reference_list ]
阅读全文