Qufang

摘要： conda install nvidia/label/cuda-11.8.0::cuda-nvcc 阅读全文

摘要： 1. torch中的索引矩阵 torch中有很多场景下都会生成索引矩阵,索引矩阵的shape和tensor的shape是相同的 a_tensor, a_index = torch.topk(a, dim=1) # or a_tensor, a_index = torch.sort(a, dim=1) 阅读全文

摘要： 1/287 [04:08<19:42:56, 248.17s/it, loss=0.710, v_num=60] 时间一样的数字是类似于tqdm的东西，第一个是所用时间，第二个是估计剩余时间ETA，第三个是一次迭代所需时间, 第四个是log的版本没啥影响 阅读全文

摘要： pickle中可以存储list, tensor, array类型的数据 但是tensor占的内存要比list和array大的多 经测试, 存储nuscene mini数据集的参数信息, tensor: 39M array: 23 M list: 22M 参考其他博文了解到, 存储tensor直接读取 阅读全文