• Sepehr Sameni's avatar
    make auxiliary heads in pretrained models optional (#828) · 50ea596e
    Sepehr Sameni authored
    * add aux_logits support to inception
    
    it is related to pytorch/pytorch#18668
    
    * instantiate InceptionAux only when requested
    
    it is related to pytorch/pytorch#18668
    
    * revert googlenet
    
    * support and aux_logits in pretrained models
    
    * return namedtuple when aux_logit is True
    50ea596e
inception.py 12.9 KB