第七章 PyTorch 小技巧汇总

• 第七章 PyTorch 小技巧汇总
  • 7.1 模型保存与加载
  • 7.2 Finetune 模型微调
  • 7.3 GPU使用
  • 7.4 模型训练代码模板
  • 7.5 TorchMetrics 模型评估指标库
  • 7.6 Albumentations 数据增强库
  • 7.7 TorchEnsemble 模型集成库

第七章简介

本章介绍开发过程中常用的代码段、工具模块和技巧等，初步设计有模型保存与加载、模型Finetune技巧、GPU使用技巧、训练代码框架等。

本章小结会续更新，将工作中遇到的小技巧分享出来。