Exemple #1
0
        private void SyncParamsFromDevices()
        {
            _exec_group.GetParams(_arg_params, _aux_params);
            if (_kvstore != null && _update_on_kvstore.HasValue && _update_on_kvstore.Value)
            {
                foreach (var p in _arg_params)
                {
                    if (p.Value.SType == StorageStype.RowSparse)
                    {
                        var row_ids = nd.Arange(0, p.Value.Shape[0], dtype: DType.Int64);
                        _kvstore.RowSparsePull(p.Key, p.Value, row_ids: row_ids);
                    }
                }
            }

            _params_dirty = false;
        }