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.
In train.py
main()function, instead ofval_triplet_sampler = davis.TripletSampler(dataset=val_data_source, num_triplets=args.num_val_batches, randomize=True)must beval_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.