Created by: ceci3
depend paddleslim changes in https://github.com/PaddlePaddle/PaddleSlim/pull/365
GAN compression demo dir：
|—— gan_compression.py 入口文件，封装的接口，用户可以直接调用接口开始整个流程 | |—— dataset 数据目录 |————|———— data_reader.py |————|———— data_loader.py | |—— models 模型目录，包含生成网络，判别网络和基础的一些模块 |————| |————|——— generator 生成网络目录 |————|—————|———— resnet_generator.py 基于resnet block的生成器 |————|—————|———— mobile_generator.py 基于 mobile resnet block的生成器 |————|—————|———— super_generator.py 基于 mobile resnet block的支持OFA格式的生成器 |————|—————|———— sub_generator.py 基于 mobile resnet block通过传入config构造生成器 |————|——— base_model.py 基础模型 |————|——— cyclegan_model.py cyclegan 模型 |————|——— discriminator.py 判别网络文件 |————|——— loss.py loss文件 |————|——— network.py 构造具体网络结构的文件 |————|——— test_model.py 测试模型 | |————|——— distillers 蒸馏目录 |————|————|———— base_resnet_distiller.py 基础蒸馏结构 |————|————|———— resnet_distiller.py resnet格式的蒸馏逻辑 | |————|——— supernents |————|————|———— resnet_supernet.py resnet格式的OFA训练逻辑 | |————|——— utils 依赖目录 |————|————|———— get_args.py 参数列表 |————|————|———— image_pool.py cyclegan pool选择 |————|————|———— optimization.py 优化器文件 |————|————|———— util.py save/load之类的接口 |————|————|———— weight_transfer.py 裁剪步骤中两个模型的参数迁移 | |————|——— configs 网络配置目录 |————|————|———— single_configs.py 单个网络结构采样 |————|————|———— resnet_configs.py resnet网络结构采样 | |————|———— README.md /README_cn.md | |————|———— export.py 导出模型 | |————|———— get_real_stat.py 计算数据集的均值方差 | |————|———— search.py 搜索超网络 | |————|———— select_arch.py 选择网络结构 | |————|———— test.py 测试网络
TODO :
- support multi_gpu