I’ll ask you one more thing.
If so, is there any other way to receive and train different image sizes and different batch sizes as inputs except collate_fn, concatdataset??
As far as I know, the batch size should be the same when concatdataset is applied.
Thank you