Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Train and Valid Data Structure #2

Open
iamrukeshduwal opened this issue Apr 2, 2024 · 5 comments
Open

Train and Valid Data Structure #2

iamrukeshduwal opened this issue Apr 2, 2024 · 5 comments

Comments

@iamrukeshduwal
Copy link

How to maintain the image folder before training the model?

@nileshkumar0726
Copy link
Owner

The train and valid folders should contain tumours as 2d images and the data loader will pair up images from that folder to be used for training.

@iamrukeshduwal
Copy link
Author

Thank you for your valuable time. Here i get some issue after I maintained the folder structure.
train.py", line 75, in fit
for src, tgt, src_img in tqdm(train_loader):
ValueError: not enough values to unpack (expected 3, got 2)

@nileshkumar0726
Copy link
Owner

For the current version, we are using a vanilla VAE so the appropriate file to use would be train_vanilla.py

@iamrukeshduwal
Copy link
Author

Still same error

@nileshkumar0726
Copy link
Owner

Could you double-check if you are using the correct function? In train_vanilla.py there are only two values required src and tgt

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants