eora-ai / torchok

Production-oriented Computer Vision models training pipeline for common tasks: classification, segmentation, detection and representationšŸ„¤
https://torchok.readthedocs.org
Apache License 2.0
54 stars 6 forks source link

Remove ability to specify multiple training dataloaders #169

Open VladVin opened 1 year ago

VladVin commented 1 year ago

Is your feature request related to a problem? Please describe. It is impossible to use multiple training datasets according to Lightning's interface (there is no dataloader_idx parameter for training_step). But TorchOk supports a list of training datasets though it won't work - the user has to specify only one training dataloader to make it work.

Describe the solution you'd like Remove the ability to pass multiple training dataloaders

Describe alternatives you've considered We can also create our own wrapper around a list of datasets, but it will be confusing for a user

Additional context