INNER CODE UNIT · Python

train_loader

NVIDIA/DeepLearningExamples · PyTorch/Detection/SSD/main.py:161

    train_loader = get_train_loader(args, args.seed - 2**31)

    val_dataset = get_val_dataset(args)
    val_dataloader = get_val_dataloader(val_dataset, args)

    ssd300 = SSD300(backbone=ResNet(backbone=args.backbone,
                                    backbone_path=args.backbone_path,
                                    weights=args.torchvision_weights_version))
    args.learning_rate = args.learning_rate * args.N_gpu * (args.batch_size / 32)
    start_epoch = 0
    iteration = 0
    loss_func = Loss(dboxes)

    if use_cuda:
        ssd300.cuda()
        loss_func.cuda()

    optimizer = torch.optim.SGD(tencent_trick(ssd300), lr=args.learning_rate,

View source record →