在PyTorch中打印模型摘要的方法

PyTorch中model.train()的作用解析