包含标签:机器学习 的文章
两百行的图像分类,用CIFAR10训练ResNet18
已阅: 6 水一下这个月的稿子,网上肯定很多,但写都写了就发出来。 # train.py from torch.utils.data import DataLoader from torch.autograd import Variable import torchvision.models as models import torchvision.transforms as T import torch.nn.functional as F import matplotlib.pyplot as plt import numpy as np import torchvision import argparse import random import torch import time import s……ViT(TransReID)模型各阶段形状
已阅: 119 之前看TransReID代码时的记录,他们代码写的很全,训练测试都有。 这段时间在这个基础上改了点代码,暂时没出现模型理解错的地方。 ViT外的改进没有记录。 有错误或疑问请留言,谢谢。代码TransReID-GitHub 下面是TransReID里的ViT结构,不包括TransReID的改进。 代码简写了。参数设置: img_size = [H, W, C] = [224, 224, 3] # 图片尺寸,代码中的img_size为[H, W] patch_size = [16, 16] # patc……
(◕ܫ◕)
评论于 MMDet实例分割loss_rpn_bbox为nan但其它loss正常的解决