摘要：
cuDNN使用非确定性算法，并且可以使用torch.backends.cudnn.enabled = False来进行禁用 如果设置为torch.backends.cudnn.enabled =True，说明设置为使用使用非确定性算法 然后再设置： 那么cuDNN使用的非确定性算法就会自动寻找最适合 阅读全文
摘要：
其实在代码的开头添加下面几句话即可： 为了生成随机数设置种子。返回一个torch.Generator对象 参数： seed (int) – 期望的种子数 为当前GPU生成随机数设置种子。如果CUDA不可用，调用该方法也是安全的；在这种情况下，该调用就会被忽略 参数： seed (int) – 期望的 阅读全文
摘要：
在为数据分类训练分类器的时候，比如猫狗分类时，我们经常会使用pytorch的ImageFolder: 使用可见pytorch torchvision.ImageFolder的使用 这里想实现的是如果想要覆写该函数，即能使用它的特性，又可以实现自己的功能 首先先分析下其源代码： ImageFolder 阅读全文