摘要：print(torch.nn.functional.one_hot(t, num_classes=7)) 有个坑，使用的时候必须转换为 torch.int64 类型，不然会报错 t = t.to(torch.int64) 阅读全文

摘要：pip安装依赖pydot和graphviz并且安装软件sudo apt install graphviz，有个坑，windows安装软件之后安装的依赖是pydot-ng 注意：模型的第一层需要把形状传进去 import os os.environ['TF_CPP_MIN_LOG_LEVEL'] = 阅读全文

摘要：前提是先安装好驱动程序，驱动程序是根据显卡的型号来选择的 1、安装CUDA Toolkit cuda10.0地址：https://link.zhihu.com/?target=https%3A//developer.nvidia.com/cuda-10.0-download-archive 2、查看 阅读全文

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