摘要:
需要导入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) 阅读全文