Okay, I’ve adapted a PyTorch VAE and the Learner seems happy enough loading it. However, I’m trying to run lr_find() on it and it’s complaining about the data. I just have two folders of images, in train and test, but I’m really not sure how I’m supposed to provide the same image as input and output (i.e., for the autoencoder)??