2022-09-20から1日間の記事一覧

loss, accuracyの計算方法まとめ

criterionのreductionはデフォルトで'mean'。len(data_loader)はバッチ数。なので、epochの平均lossの算出は、バッチごとの(平均)lossを加算していって、len(data_loader)で割ればよい。正解数は correct += pred.eq(target.view_as(pred)).sum().item() …

onnxの使い方

・pthファイルからonnxファイルへの変換 import torchfrom torch import onnxfrom torchvision import transforms, datasetsdef main(): transform = transforms.Compose([ transforms.ToTensor(), transforms.Normalize((0.5, 0.5, 0.5), (0.5, 0.5, 0.5)) …