Ranuga Disansa

Pull request : #1825

It's a great idea. It could help.

oh, ok Yes, Quick Vision is a great library.

hi, I would like to help with this issue. With best regards, Ranuga

I will start working on this. :) With best regards, Ranuga

Hi, I want to know what the issue is to use a torch model in PyTorch Lightning and Fine Tune? With best regards, Ranuga

ok, thank you @oke-aditya I will try to fix the issue.

ok thank you @oke-aditya

hi, I am currently building a demo of this and my question is I can. ```py from torchvision.models import googlenet model = googlenet().to(device) print(model) # Prints the model architecture model.fc...

hi, I usually add \_\_init\_\_ `self.output = Linear(1000,len(classes))` forward ```py preds = self.tl_model(X) preds = self.output(preds) ``` I don't know if this is the best way but when I am...