hub / github.com/Tencent/PocketFlow

github.com/Tencent/PocketFlow @main

Chat with this repo

repository ↗ · DeepWiki ↗ · + Follow

802 symbols 2,492 edges 121 files 610 documented · 76% ● updated 3y ago★ 2,90873 open issues

Browse by type

⨍ Functions 751◇ Types & classes 51

What it actually does AI analysis from the code graph — generated when you open this

loading…

Core symbols most depended-on inside this repo

↓ 25 filesReplayBuffer

Replay buffer for storing state-action-reward transitions. Each transition consists of five components: 1. state: current state …

rl_agents/ddpg/replay_buffer.py

↓ 20 filesMultiGpuWrapper

Wrapper for multi-GPU training.

utils/multi_gpu_wrapper.py

↓ 19 filesAbstractLearner

Abstract class for learners. A learner should take a ModelHelper object as input, which includes the data input pipeline and …

learners/abstract_learner.py

↓ 9 filesDistillationHelper

Helper for training with distillation loss. Other learners can use calc_loss() (remember to call initialize() first to make sure …

learners/distillation_helper.py

↓ 7 filesAbstractModelHelper

Abstract class for model helpers. A model helper should define the following function interface: 1. Data input pipeline for …

nets/abstract_model_helper.py

↓ 7 filescreate_learner

Create the learner as specified by FLAGS.learner. Args: * sm_writer: TensorFlow's summary writer * model_helper: model helper …

learners/learner_utils.py

browse all functions →

Shape

Method 447

Function 304

Class 51

Languages

Python100%

TypeScript1%

Modules by API surface

learners/channel_pruning/channel_pruner.py28 symbols

nets/vgg_at_pascalvoc.py25 symbols

learners/channel_pruning/learner.py24 symbols

learners/weight_sparsification/pr_optimizer.py23 symbols

learners/channel_pruning_rmt/learner.py23 symbols

nets/faster_rcnn_at_pascalvoc.py22 symbols

learners/channel_pruning/model_wrapper.py22 symbols

utils/external/ssd_tensorflow/preprocessing/ssd_preprocessing.py21 symbols

learners/nonuniform_quantization/utils.py21 symbols

rl_agents/ddpg/agent.py19 symbols

learners/discr_channel_pruning/learner.py18 symbols

utils/external/ssd_tensorflow/dataset/convert_tfrecords.py17 symbols

For agents

$ claude mcp add PocketFlow \
  -- python -m otcore.mcp_server <graph>

⬇ download graph artifact

github.com/Tencent/PocketFlow @main

Core symbols most depended-on inside this repo

Shape

Languages

Modules by API surface

For agents

Ask about this repo answers extend the page