pytorch查看模型参数总结
1:DNN_printer其中(3, 32, 32)是输入的大小,其他方法中的参数同理from DNN_printer import DNN_printerbatch_size = 512def train(epoch): print('\nEpoch: %d' % epoch) net.train() train_loss = 0 correct = 0 total = 0 // put the code here and you can get
原创
2021-07-06 14:28:48 ·
2912 阅读 ·
0 评论