Yes, I am!
if params.is_train:
    print("MODE: Training")
    #----create trainloader
    trainset     = high_dimensional_dataset(params.data, params.train.lIndex, params.train.hIndex, params.train.aug, params.train.get_info) #--make sure that get_info is false
    trainloader  = DataLoader(trainset, batch_size=params.train.batch_size, shuffle=True, num_workers=params.train.num_workers) #-----shuffle
    train_batches = int(len(trainset)/params.train.batch_size)  (=4096*realizations/batch_size)
    print("Trainloader created.")

    #----create validloader
    validset     = high_dimensional_dataset(params.data, params.valid.lIndex, params.valid.hIndex, params.valid.aug,params.valid.get_info) #---AGAIN, get_info should be false
    validloader  = DataLoader(validset, batch_size=params.valid.batch_size, shuffle=True, num_workers=params.valid.num_workers) #----shuffle
    print("Validloader created.")

    #----train
    train(model, trainloader, validloader, myLoss, params)