-
Notifications
You must be signed in to change notification settings - Fork 5
自定义 图像预处理 | 损失函数 | 优化器 | 学习率调度器
Mr.Li edited this page Jul 21, 2022
·
7 revisions
- 入口
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"
。