Logging updated in loading

[lrm22005]

I found a error with a file based on the size, I like to ensure that this error is not associated with our data.

The error was in
_, labeled_data, _, _, labeled_labels = load_data(data_path, labels_path, dataset_size=141, train=True, data_format=data_format, return_all="labeled")

and it is associated with the size of the data readed

                time_freq_plot = np.array(pd.read_csv(seg_path, header=None))
                time_freq_tensor = torch.Tensor(time_freq_plot).reshape(1, 128, 128)