Custom data loader for 3D data

You are correct. If you create a WideResnet Pytorch model, then you should be able to input that into the Learner class like I did in the example in my last post. However, you will also (probably) have to create a custom dataset class which you can then instantiate with a DataBunch fastai class.

To say that all in a different way, if you have your pytorch WideResnet model in a variable named model and you create a DataBunch class instance called data with your custom dataset. Then you should be able to call

learner = Learner(data, model, loss_func=..., metrics=[...])

and have a working fastai learner which supports the .fit, .find_lr, etc. Someone can (please) correct me if I’m wrong about any of what I said, or chime in if there is an easier way to do this.

1 Like