chirp

摘要： 使用 .numel() 方法可以统计模型的参数量。 以下代码摘自 In-context Autoencoder (ICAE) 的代码仓库。输入 nn.Module，统计该模型的参数量和可训练参数量： def print_trainable_parameters(model): trainable_p 阅读全文