pytorch
FengGHT
这个作者很懒,什么都没留下…
展开
-
tieredImageNet
出自:Meta-learning for Semi-supervised Few-shot Classificationfeature:1. subset of ILSVRC-122. 608 classes3. broader categories:34 categories in total, each containing 10 ~ 30 classes4. 20 training, 6 val, 8 testing5.原创 2020-08-13 14:36:39 · 923 阅读 · 0 评论 -
mmdet错误
mmdetection loss=Nan 或者device-side assert triggered错误1.权重文件的版本问题2.权重文件的head部分修改造成的3.学习率太大原创 2020-08-03 11:10:04 · 381 阅读 · 0 评论 -
pytorch 记录
这里写自定义目录标题pytorch 加载部分权重pytorch 固定权重pytorch 加载部分权重pretrained_dict = torch.load('{0}/{1}'.format(opt.outf, opt.resume_posenet))model_dict = model.state_dict()#筛选条件pretrained_dict = {k: v for k, v in pretrained_dict.items() if k in model_dict}model_di原创 2020-06-29 19:07:45 · 122 阅读 · 0 评论