PyTorch
PyTorch有关内容
月光哒哒哒
技术改变世界
展开
-
Anaconda中修改PyTorch下载预训练模型的默认保存路径
今天下载vgg16预训练模型时发现默认下载位置在C盘,这对C盘空间小的同学很不友好。下面介绍修改预训练模型默认保存路径的方法(其他模型同样操作)。PyTorch的默认下载路径由load_state_dict_from_url()函数确定,因此找到该函数进行修改即可。按Ctrl+鼠标左键进入vgg.py源码,搜索load_state_dict_from_url,进入utils.py进入hub.py搜索load_state_dict_from_url将model_dir = None换成mod原创 2022-05-26 22:03:10 · 2288 阅读 · 1 评论 -
Python __getitem__()方法
如果在python的类中定义了__getitem__()的魔法方法,实例对象可以通过键值来取值。 class Test: def __init__(self, num, flower): self.num = num self.flower = flower self.dict = dict() self.getDict() def getD...转载 2022-05-21 15:23:00 · 314 阅读 · 0 评论 -
python类中的 __call__()方法详解
<p><code>__call__()</code>是一种magic method,在类中实现这一方法可以使该类的实例(对象)像函数一样被调用。默认情况下该方法在类中是没有被实现的。使用callable()方法可以判断某对象是否可以被调用。</p> __call__()方法的作用其实是把一个类的实例化对象变成了可调用对象,也就是说把一个类的实例化对象变成了可调用对象,只要...转载 2022-05-21 00:07:36 · 13816 阅读 · 1 评论