Misplaced parameter

[marcelo-mendonca]

In train.py main() function, instead of val_triplet_sampler = davis.TripletSampler(dataset=val_data_source, num_triplets=args.num_val_batches, randomize=True)
must be val_triplet_sampler = davis.TripletSampler(dataset=val_data_source, num_triplets=args.val_batch_size, randomize=True) to avoid throwing an exception due to incompatible tensor dimensions.