Pytorch 中的 memory_format 参数 与 channels_last
Pytorch 中的 memory_format 参数
·
-
memory_format(torch.memory_format, optional)：期望返回的tensor的内存格式，默认为torch.preserve_format.
memory_format = torch.channels_last-
PyTorch 1.5 引入了channels_last对卷积网络内存格式的支持。此格式旨在与 AMP结合使用，以进一步加速具有 Tensor Cores的卷积神经网络
有“AI”的1024 = 2048，欢迎大家加入2048 AI社区
更多推荐
1
0- 0
已为社区贡献3条内容
所有评论(0)