12 2021 档案

pytorch查看模型信息：torchinfo
摘要：查看模型流程、tensor的变化、参数量 example： from torchinfo import summary for X, y in train_dl: print(summary(model, X.shape)) break output： Layer (type:depth-idx) 阅读全文
posted @ 2021-12-02 10:03 iFzh 阅读(1869) 评论(0) 推荐(0)