它還可以計算引數的數量和列印給定網路的每層計算成本。 支援layer:Conv1d/2d/3d,ConvTranspose2d,BatchNorm1d/2d/3d,啟用(ReLU, PReLU, ELU, ...
確定! 回上一頁