再下面的 if 语句的作用简单理解就是,如果 pin_memory=True ,那么Pytorch会采取一系列操作把数据拷贝到GPU,总之就是为了加速。 综上可以知道DataLoader ...
確定! 回上一頁