huantaiqiu的博客

原创 关于动态调整学习率

import torch import torch.nn as nn from torch.optim.lr_scheduler import LambdaLR initial_lr = 0.1 class model(nn.Module): def __init__(self): super().__init__() self.conv1 = nn.Conv2d(in_channels=3, out_channels=3, kernel_size=3)