2022-09-20から1日間の記事一覧
criterionのreductionはデフォルトで'mean'。len(data_loader)はバッチ数。なので、epochの平均lossの算出は、バッチごとの(平均)lossを加算していって、len(data_loader)で割ればよい。正解数は correct += pred.eq(target.view_as(pred)).sum().item() …
・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)) …