摘要:一、继承nn.Module类并自定义层 我们要利用pytorch提供的很多便利的方法,则需要将很多自定义操作封装成nn.Module类。 首先,简单实现一个Mylinear类: from torch import nn # Mylinear继承Module class Mylinear(nn.Mod
阅读全文
摘要:1.使用keras做mnist分类时,运行时GPU报错 错误信息如下: 2019-07-06 10:26:32.949617: I T:\src\github\tensorflow\tensorflow\core\platform\cpu_feature_guard.cc:141] Your CPU
阅读全文