
pytorch
是晓Xiao哦
这个作者很懒,什么都没留下…
展开
-
pytorch:optim.lr_scheduler.ReduceLROnPlateau
1、调整学习率torch.optim.lr_scheduler.ReduceLROnPlateau allows dynamic learning rate reducing based on some validation measurements.2、主要参数mode(str)——有两种模式min和max,min表示指标不再降低,max表示指标停止增加factor(float)——学习率降低的因素,new_lr = lr * factor,默认值为0.1patient(int)——有多少个e原创 2020-06-13 16:50:09 · 1873 阅读 · 0 评论 -
pytorch 转置卷积
ConvTranspose2d代码参数: class torch.nn.ConvTranspose2d(in_channels, out_channels, kernel_size, stride=1, padding=0, output_padding=0, bias=True)原创 2020-05-21 15:25:50 · 238 阅读 · 0 评论 -
permute函数
permute函数:可以同时多次交换tensor的维度示例:b = a.permute(0,2 ,1) 将a的维度索引1和维度索引2交换位置原创 2020-05-21 15:09:41 · 22413 阅读 · 3 评论 -
unsqueeze函数
unsqueeze:增加维度操作的函数b = a.unsqueeze(0) #在a原来维度索引0之前增加一个维度原创 2020-05-21 14:58:46 · 768 阅读 · 0 评论