• Patrick Wieschollek's avatar
    Add ignore to SaverRestore (#302) · 6be159d3
    Patrick Wieschollek authored
    * Add ignore to SaverRestore
    
    When fine-tuning a model, one might want to skip some tensors from
    restoring.
    
    The following example will not load the learning rate from the
    checkpoint file:
    
      config.session_init = SaverRestore(args.load,
          ignore=['learning_rate'])
    
    * Update sessinit.py
    6be159d3
sessinit.py 8.37 KB