[Enhancement Inference + Finetuning Functionality]

[Jacobsolawetz]

Enhancement

Perhaps this functionality already exists, but I was thinking it would be nice to have AutoModel have something like .finetune(dataset_location) and .infer(input_img)

[FrancescoSaverioZuppichini]

Hi @Jacobsolawetz unfortunately the aim of glasses is just to provide well-written models easy to use but the logic behind how to train them is up to you. I could add a binding to pytorch-lighting to make it easier to finetune them thou. What do you think?