pytorch
哎呦不错的温jay
关注研究数字人、元宇宙、计算机视觉、图形学、AIGC、语言大模型。
展开
-
RuntimeError: Attempting to deserialize object on CUDA device 2 but torch.cuda.device_count() is 2.
解决报错RuntimeError: Attempting to deserialize object on CUDA device 2 but torch.cuda.device_count() is 2. Please use torch.load with map_location to map your storages to an existing device.原创 2021-12-31 16:21:40 · 10902 阅读 · 1 评论 -
Pytorch——将模型load到gpu或cpu上
很多时候我们在gpu上训练一个模型,但是在inference的时候不想使用gpu。或者想在别的gpu上使用,那么怎么办呢?需要在load的时候就选择device。保存了模型的参数(model.state_dict())到文件model.pth中。1、cpu->cpu 或gpu->gpu这种情况是最简单的:checkpoint = torch.load('model...原创 2019-11-15 17:49:40 · 6590 阅读 · 0 评论 -
Pytorch——torch.nn.functional.interpolate函数
最近写pytorch的时候用到了这个函数:torch.nn.functional.interpolate一时没太懂这个函数是干嘛的,所以看了下pytorch的官方文档:torch.nn.functional.interpolate(input,size=None,scale_factor=None,mode='nearest',align_corners=None):Down/...原创 2019-09-24 17:51:35 · 41549 阅读 · 3 评论 -
Pytorch——tensor.expand_as()函数示例
tensor.expand_as()这个函数就是把一个tensor变成和函数括号内一样形状的tensor,用法与expand()类似。差别是expand括号里为size,expand_as括号里为其他tensor。举例:import torchx = torch.tensor([[1], [2], [3]])print('xsize:',x.size())print('x:',...原创 2019-09-24 19:21:56 · 16969 阅读 · 1 评论