Training with BIVA
Opened this issue · 0 comments
Davidyao99 commented
Hi, I read your paper and it is very insightful. I am very excited to use your model and wish to train it on some custom dataset!
Do you have any suggestions on how I should proceed? I am trying to remove your DataModule class and use the pytorch ImageFolder to generate the necessary dataloaders.
Additionally, I am hoping to use the BIVA architecture. What should I set as my config_deterministic and config_stochastic arguments?
Thank you so much for your time and really looking forward to making this work! ;)