Esempio n. 1
0
        public Saver(IVariableV1[] var_list = null,
                     bool reshape           = false,
                     bool sharded           = false,
                     int max_to_keep        = 5,
                     float keep_checkpoint_every_n_hours = 10000,
                     string name = null,
                     bool restore_sequentially = false,
                     SaverDef saver_def        = null,
                     ISaverBuilder builder     = null,
                     bool defer_build          = false,
                     bool allow_empty          = false,
                     SaverDef.Types.CheckpointFormatVersion write_version = SaverDef.Types.CheckpointFormatVersion.V2,
                     bool pad_step_number     = false,
                     bool save_relative_paths = false,
                     string filename          = "")
        {
            _var_list    = var_list;
            _reshape     = reshape;
            _sharded     = sharded;
            _max_to_keep = max_to_keep;
            _keep_checkpoint_every_n_hours = keep_checkpoint_every_n_hours;
            _name = name;
            _restore_sequentially = restore_sequentially;
            _saver_def            = saver_def;
            _builder         = builder;
            _is_built        = false;
            _allow_empty     = allow_empty;
            _write_version   = write_version;
            _pad_step_number = pad_step_number;

            if (!defer_build)
            {
                build();
            }
            if (_saver_def != null)
            {
                _check_saver_def();
                _write_version = _saver_def.Version;
            }

            _save_relative_paths  = save_relative_paths;
            _object_restore_saver = null;

            _last_checkpoints          = new Dictionary <string, float>();
            _checkpoints_to_be_deleted = new Dictionary <string, float>();
        }
Esempio n. 2
0
 public BaseSaverBuilder(SaverDef.Types.CheckpointFormatVersion write_version = SaverDef.Types.CheckpointFormatVersion.V2)
 {
     _write_version = write_version;
 }
Esempio n. 3
0
 public BulkSaverBuilder(SaverDef.Types.CheckpointFormatVersion write_version = SaverDef.Types.CheckpointFormatVersion.V2) : base(write_version)
 {
 }