01 2021 档案

pytorch one hot
摘要：print(torch.nn.functional.one_hot(t, num_classes=7)) 有个坑，使用的时候必须转换为 torch.int64 类型，不然会报错 t = t.to(torch.int64) 阅读全文
posted @ 2021-01-20 16:48 consolexinhun 阅读(348) 评论(0) 推荐(0)
TF2 可视化Loss函数，并且导出模型图
摘要：pip安装依赖pydot和graphviz并且安装软件sudo apt install graphviz，有个坑，windows安装软件之后安装的依赖是pydot-ng 注意：模型的第一层需要把形状传进去 import os os.environ['TF_CPP_MIN_LOG_LEVEL'] = 阅读全文
posted @ 2021-01-18 15:27 consolexinhun 阅读(236) 评论(0) 推荐(0)
ubuntu安装cuda10.0
摘要：前提是先安装好驱动程序，驱动程序是根据显卡的型号来选择的 1、安装CUDA Toolkit cuda10.0地址：https://link.zhihu.com/?target=https%3A//developer.nvidia.com/cuda-10.0-download-archive 2、查看 阅读全文
posted @ 2021-01-16 21:05 consolexinhun 阅读(731) 评论(0) 推荐(0)
git clone 加速
摘要：方法一：改 hosts 文件 缺点：提速不明显，而且过一段时间没用了 方法二：从码云中按照 Github 仓库的 Https 地址导入项目，然后通过码云来 clone 缺点：操作步骤繁杂 方法三：在 github.com 后面加上 .cnpmjs.org 或者换个域名 github.com 替换成 阅读全文
posted @ 2021-01-12 20:12 consolexinhun 阅读(151) 评论(1) 推荐(0)
pytorch中画模型图
摘要：需要导入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 阅读(465) 评论(0) 推荐(0)
torchsummary使用
摘要：先导包，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 阅读(1458) 评论(0) 推荐(0)