totttteのブックマーク - はてなブックマーク

[PyTorch]modelとdatasetがcudaモードになっているか確認する - Qiita
はじめに PyTorchはテンソルに対してhoge.to(device)などで簡単にcpuとgpuモードを切り替えられますが，よくこのデータセットやモデルがcpuかgpuなのかわからなくなったので，確認する方法を書き残しときます．確認方法前提としてデータセットとモデルの準備は IMAGE_SIZE=224 BATCH_SIZE=20 TRAIN = 'train' DATA_DIR = 'dataset/predata/' DEVICE = torch.device("cuda:0" if torch.cuda.is_available() else "cpu") data_transf orms = { TRAIN: transf orms.Compose([ transf orms.Resize(IMAGE_SIZE), transf orms.ToTensor(), ]) } # データ
totttte 2024/04/27
リンク
[Pytorch] torch.catの動きを理解する - Qiita
はじめに PyTorchの公式(1)にあるtorch.catの例だけでは良く分からなかったので他の例を使って理解するための書き残しです． sample code import torch input1 = torch.randn(2, 3, 4) input2 = torch.randn(2, 3, 4) input3 = torch.randn(5, 3, 4) input_list = [input1, input2, input3] output1 = torch.cat(input_list, dim=0) print(output1.size()) # torch.Size([9, 3, 4]) output2 = torch.cat(input_list, dim=1) # error print(output2.size()) output3 = torch.cat(inpu
totttte 2024/04/24
リンク
1

はてなブックマーク