SwAV 코드 살펴보기
코드 : facebookresearch/swav def train(train_loader, model, optimizer, epoch, lr_schedule, queue): batch_time = AverageMeter() data_time = AverageMeter() losses = AverageMeter() softmax = nn.Softmax(dim=1).cuda() model.train() use_the_queue = False end = time.time() for it, inputs in enumerate(train_loader): # measure data loading time data_time.update(time.time() - end) # update learning rate ite..
2021.03.23