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

How to train with single modality data? #4

Open
QL111111 opened this issue Jul 16, 2019 · 1 comment
Open

How to train with single modality data? #4

QL111111 opened this issue Jul 16, 2019 · 1 comment

Comments

@QL111111
Copy link

Thank you for sharing this code.
I want to train on a single modality dataset. Do you have any idea how should I change this code?

@sacmehta
Copy link
Owner

sacmehta commented Jul 17, 2019

We concatenate four modailities into 1, see here.

input = torch.cat([inp, inputA, inputB, inputC], 1) # dim-0 is batch

Instead you can feed one modality.

ALso, you need to change number of channels to 1. We use 4 for four modalities.

args.channels = 4 # because 4 modalities. You can think each modality as a single channel (R or G or B) of an RGB image

Obviously, you need to make changes in other files too.

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