它還可以計算參數的數量和打印給定網絡的每層計算成本。 支持layer:Conv1d/2d/3d,ConvTranspose2d,BatchNorm1d/2d/3d,激活(ReLU, PReLU, ELU, ...
確定! 回上一頁