Massive changes, using better dataset and now returning random masks 3922a86 smhavens commited on Dec 9, 2023
Bug found: train_data was used for loop and must be everything 7bf917c smhavens commited on Dec 7, 2023