consolexinhun

摘要： 需要导入tensorboard包 pip install tensoboard 代码 from torch.utils.tensorboard import SummaryWriter model = FPN() with SummaryWriter(comment="FPN") as w: w.a 阅读全文

摘要： 先导包，pip install torchsummary 然后 from torchsummary import summary model = FPN() y = summary(model, (3, 224, 224), device="cpu") print(y) 阅读全文