Skip to content
Snippets Groups Projects
  • Neta Zmora's avatar
    ba653d9a
    Check if correct version of PyTorch is installed. · ba653d9a
    Neta Zmora authored
    The 'master' branch now uses PyTorch 0.4, which has API changes that
    are not backward compatible with PyTorch 0.3.
    
    After we've upgraded Distiller's internal implementation to be
    compatible with PyTorch 0.4, we've added a check that you are using
    the correct PyTorch version.
    
    Note that we only perform this check in the sample image classifier
    compression application.
    ba653d9a
    History
    Check if correct version of PyTorch is installed.
    Neta Zmora authored
    The 'master' branch now uses PyTorch 0.4, which has API changes that
    are not backward compatible with PyTorch 0.3.
    
    After we've upgraded Distiller's internal implementation to be
    compatible with PyTorch 0.4, we've added a check that you are using
    the correct PyTorch version.
    
    Note that we only perform this check in the sample image classifier
    compression application.