optimizer = torch.optim.Adam(model.parameters(), lr=1, weight_decay=0.0001) #
scheduler = ReduceLROnPlateau(optimizer, mode='min', factor=0.8,
欢迎分享,转载请注明来源:内存溢出
optimizer = torch.optim.Adam(model.parameters(), lr=1, weight_decay=0.0001) #
scheduler = ReduceLROnPlateau(optimizer, mode='min', factor=0.8,
欢迎分享,转载请注明来源:内存溢出
评论列表(0条)