自定义 图像预处理 | 损失函数 | 优化器 | 学习率调度器 - bobo0810/Classification GitHub Wiki

图像预处理

1. 入口DataSets/preprocess.py 具体实现,命名为MyProcess。
2. 启用: 修改config.py Process = "MyProcess"。

损失函数

1. 入口Models/Loss 具体实现,命名为MyLoss。
2. 启用: 修改config.py Loss = "MyLoss"。

优化器

1. 入口Models/Optimizer 具体实现,命名为MyOptimizer。
2. 启用: 修改config.py Optimizer = "MyOptimizer"。

学习率调度器

1. 入口Models/Scheduler/具体实现

@SCHEDULER_REGISTRY.register()
def MyScheduler(epochs, optimizer):
   xxxxx

2. 启用: 修改config.py Scheduler = "MyScheduler"。