Skip to content
Snippets Groups Projects
  • Neta Zmora's avatar
    11402988
    compress_classifier.py: add an option to load a model in serialized mode · 11402988
    Neta Zmora authored
    By default, when we create a model we  wrap it with DataParallel to benefit
    from data-parallelism across GPUs (mainly for convolution layers).
    
    But sometimes we don't want the sample application to do this: for
    example when we receive a model that was trained serially.
    This commit adds a new argument to the application to prevent
    the use of DataParallel.
    11402988
    History
    compress_classifier.py: add an option to load a model in serialized mode
    Neta Zmora authored
    By default, when we create a model we  wrap it with DataParallel to benefit
    from data-parallelism across GPUs (mainly for convolution layers).
    
    But sometimes we don't want the sample application to do this: for
    example when we receive a model that was trained serially.
    This commit adds a new argument to the application to prevent
    the use of DataParallel.