Converting Model to different input and output channels

[lelouch0204]

Hi,

I found your work highly interesting and was taking a look at the model. I wanted to make it so that domain A has 3 input channels and B has 31 channels, but could not find a way to do so. Could you suggest where I could make the changes?

[Rivendile]

Thanks for your interest. I think there are two steps. First, change the input_dim / output_dim in config file. Second, modify the architecture of the network in networks.py.