search

darsnack / FluxModels.jl

Standard (pre-trained) ML models written in Flux.jl
MIT License
3 stars 5 forks source link

add googlenet #18

Closed dnabanita7 closed 4 years ago

dnabanita7 commented 4 years ago

@darsnack I have added the layers googlenet according to the paper. But I saw addition of some auxillary inception blocks in torchvision. I will go through them and check if that's necessary to add.

Edit : The auxillary inception block is used only if pre-trained is false.

dnabanita7 commented 4 years ago

I will check locally what is breaking after JuliaCon talks.

dnabanita7 commented 4 years ago

convfilter needs to be converted to Conv.