f3a62a8080bab4c5482887be46f54734f23e0b98,evaluate.py,,,#,38

Before Change

    test_dataset = test_data_loader.dataset

    // Model
    model = build_model()

    // Load checkpoint
    print("Load checkpoint from {}".format(checkpoint_path))
    checkpoint = torch.load(checkpoint_path)

After Change

    test_dataset = test_data_loader.dataset

    // Model
    model = build_model().to(device)

    // Load checkpoint
    print("Load checkpoint from {}".format(checkpoint_path))
    checkpoint = torch.load(checkpoint_path)

Instances