ExplainableML-Vision icon indicating copy to clipboard operation
ExplainableML-Vision copied to clipboard

model_name=="inception"

Open Harry-KIT opened this issue 3 years ago • 7 comments

Hi @marvinbuss , good job. I have one question. Does the below model_name always have to be inception? I do not understand why it is like that?

is_inception=(model_name=="inception")

Harry-KIT avatar Dec 13 '21 08:12 Harry-KIT

Hi @Harry-KIT, Thanks for your question! You can use any of the models as baseline and you can also extend this with newer models that are available: image

The inception model has a different architecture and needs to be treated slightly different compared to other models like ResNet. Inception uses auxiliary classifiers, which need to be handled in the output of the model. This is why there is a special check whether an inception based model is used.

Today, I would probably code this a bit nicer. Happy to accept any PRs from your side to improve the notebook. Does that answer your question?

marvinbuss avatar Dec 13 '21 08:12 marvinbuss

I got it Thank you @marvinbuss ^^.

Harry-KIT avatar Dec 14 '21 05:12 Harry-KIT

Happy to help! You are welcome!

marvinbuss avatar Dec 14 '21 08:12 marvinbuss

Hi @marvinbuss Do you have a future plan to make multi-gpus training parse on this repo?

Harry-KIT avatar Dec 15 '21 00:12 Harry-KIT

Hi @Harry-KIT, I had no plans to add this at this point in time. Would you be willing to contribute and add this?

marvinbuss avatar Dec 20 '21 08:12 marvinbuss

https://user-images.githubusercontent.com/57295863/182060895-2d972201-d876-4a29-afe6-20d44a3222e6.mp4

https://github.com/Harry-KIT/Image-Classification-on-small-datasets-in-Pytorch Hi @marvinbuss. I created similar repo based on your github repo and some other useful sources. it includes multi-gpu option as well. Thank you for that you inspired me to learn many things from your repo.

Harry-KIT avatar Aug 01 '22 02:08 Harry-KIT

Hi @Harry-KIT, Sorry for my very late response here. This is awesome. Great work and thanks for sharing! 👍

marvinbuss avatar Oct 19 '22 08:10 marvinbuss