深度学习
qi_sama
这个作者很懒,什么都没留下…
展开
-
RuntimeError: “nll_loss_forward_reduce_cuda_kernel_2d_index“ not implemented for ‘Int‘解决办法
运行pytorch在计算cross_entropy的loss遇到报错RuntimeError: "nll_loss_forward_reduce_cuda_kernel_2d_index" not implemented for 'Int'`原因是pytorch自带的损失计算函数不支持原本的标签1,2,3,4…,n,需要转换成网络需要的one_hot编码才行具体代码如下 one_hot = torch.zeros(np.array(batch_size, num_class, device=tor原创 2022-01-10 05:42:51 · 14425 阅读 · 8 评论 -
Tensorflow,Keras使用动态显存分配
Tensorflow,Keras使用动态显存分配使用tf,keras训练时常常占满显存,因此采用动态显存分配策略节约资源import tensorflow as tf #这里如果用到tf2版本,则替换为> import tensorflow.compat.v1 as tfconfig = tf.ConfigProto()config.gpu_options.allow_growth=Truesess = tf.Session(config=config)...原创 2021-12-15 06:16:02 · 2227 阅读 · 0 评论 -
Anaconda安装pytorch遇到CondaError: Downloaded bytes did not match Content-Length
官网源安装pytorch时很慢,出现Downloaded bytes did not match Content-Length下载好包之后,使用conda的离线安装就可以了conda install --offline pytorch-1.8.1-py3.9_cuda11.1_cudnn8_0.tar.bz2安装好pytorch包之后,继续执行之前的操作完成安装conda install pytorch torchvision torchaudio cudatoolkit=11.1 -c py原创 2021-03-27 17:13:05 · 2889 阅读 · 0 评论