Printing the layers of the pytorch resnet will yield: (fc): Linear(in_features=2048, out_features=1000, bias=True).
確定! 回上一頁