自定义 图像预处理 | 损失函数 | 优化器 | 学习率调度器 - bobo0810/Classification GitHub Wiki
图像预处理
- 入口
DataSets/preprocess.py
具体实现,命名为MyProcess
。 - 启用: 修改config.py
Process = "MyProcess"
。
损失函数
- 入口
Models/Loss
具体实现,命名为MyLoss
。 - 启用: 修改config.py
Loss = "MyLoss"
。
优化器
- 入口
Models/Optimizer
具体实现,命名为MyOptimizer
。 - 启用: 修改config.py
Optimizer = "MyOptimizer"
。
学习率调度器
- 入口
Models/Scheduler/
具体实现
@SCHEDULER_REGISTRY.register()
def MyScheduler(epochs, optimizer):
xxxxx
- 启用: 修改config.py
Scheduler = "MyScheduler"
。