Loop when trying to train model

[refrantz]

Hello, I'm trying to train your model on another dataset and after caching the dataset it starts a loop and starts caching the dataset again. Here is an example of what I'm talking about:

This also happens when downloading the amos22 dataset and trying to train on that (don't have the screenshot but it's literally the same problem)

Any help is appreciated, thanks in advance.

[leeh43]

Hi, thank you your interest towards our work. I am sure that our work is working, as I have tried so many times, and also only a paragraph of code is for dataloader as follows:

Have you changed any code in main_train.py?

[refrantz]

I haven't, only changed code in load_datasets_transforms.py. Am I correct in understanding that to add my dataset I need to include it with the correct folder structure in root_dir, and edit load_datasets_transforms.py so it sets the number of output channels correctly and applies the transforms?

I'll keep trying, if I'm adding the datasets correctly it's probably something with my enviroment. Thanks.

[leeh43]

Seems like you are using windows terminal to train it. Do you mind to switch to WSL2 to train it? Maybe there are some incompatibilities behind that, as my code is written under ubuntu system.

But this phenomenon is pretty weird I can say, and I think that you are doing correctly, only changing the load_datasets_transforms.py and haven't changed anything in main_train.py.

[refrantz]

That actually worked, seems like it was something with windows indeed. Thanks for the help.