7fffa98b9166a03b4a53fb40202d97b09e8e9036,train.py,,,#,21

Before Change

        targets = Variable(targets.type(Tensor), requires_grad=False)
        optimizer.zero_grad()

        loss = model(imgs, targets)

        loss.backward()
        optimizer.step()

After Change

// Get hyper parameters
hyperparams     = parse_model_config(opt.model_config_path)[0]
batch_size      = int(hyperparams["batch"])
subdivisions    = int(hyperparams["subdivisions"])
sub_batch       = batch_size // subdivisions
learning_rate   = float(hyperparams["learning_rate"])
momentum        = float(hyperparams["momentum"])

Instances