摘要: 需要导入tensorboard包 pip install tensoboard 代码 from torch.utils.tensorboard import SummaryWriter model = FPN() with SummaryWriter(comment="FPN") as w: w.a 阅读全文
posted @ 2021-01-10 17:44 consolexinhun 阅读(425) 评论(0) 推荐(0) 编辑
摘要: 先导包,pip install torchsummary 然后 from torchsummary import summary model = FPN() y = summary(model, (3, 224, 224), device="cpu") print(y) 阅读全文
posted @ 2021-01-10 17:23 consolexinhun 阅读(1382) 评论(0) 推荐(0) 编辑