チラ裏備忘録

情報整理

2021-02-01から1ヶ月間の記事一覧

nn.○○とnn.functional.○○の違い

qiita.com nn.ReLU()は、nn.Moduleを作ります。つまり、nn.Sequential()に追加できます。反対にnn.functional.reluは、relu関数を呼ぶだけで、もし記述するならforward methodの方に書きます。好きな方を使えば良いみたいですが、個人的には、print(model)と…

Adaptive_avg_pool2dとavg_pool2dの違い

discuss.pytorch.org test_input = Image.open('test_image.jpeg') # 1024x768 plt.figure(figsize=(12,4)) # avg_pool2d outputs1 = F.avg_pool2d(torchvision.transforms.ToTensor()(test_input), kernel_size=(4, 4), stride=2, padding=1) print(outputs…