随笔分类 - pytorch
摘要：一、 二、学习率更新 1、官方函数 1）milestones：lr改变时的epoch数目 2）gamma：学习率调整倍数，默认为0.1,即下降10倍 3）last_epoch：别人解释是已经运行的epoch数量！所以下一次更新lr的时候是milestones中值减去last_epoch torch.
阅读全文
摘要：明明windows10环境变量中设置了路径，还是报了这个错！需要将dll拷贝过来
阅读全文
摘要：Tensor.detach()：返回新的tensor，与当前graph分离。不可访问梯度。不过，又说与源tensor共享存储，而且修改其中之一，会触发正确性检查的错误。对返回张量的修改，不会改变原tensor，而是触发错误。 1、增加一维 torch.zeros_like(gxy)[None]
阅读全文
摘要：如果是在docker中安装，步骤如下： （1）docker pull pytorch/pytorch:1.8.1-cuda10.2-cudnn7-devel。这里必须是-devel后缀！因为编译时需要NVCC，只有-devel中有NVCC。cuda11.2有问题。 （2）按着NVIDIA/apex中
阅读全文
摘要：https://pytorch.org/tutorials/advanced/super_resolution_with_onnxruntime.html 1 定义模型 跟一般模型定义并无区别，需要torch_model.eval()或者torch_model.train(False)将模型转换为推
阅读全文
