Skip to content

Instantly share code, notes, and snippets.

View Eric-mingjie's full-sized avatar

Mingjie Sun Eric-mingjie

View GitHub Profile
# @article{ThiNet_ICCV17,
# Author = {Jian-Hao Luo, Jianxin Wu, and Weiyao Lin},
# Title = {ThiNet: A Filter Level Pruning Method for Deep Neural Network Compression},
# Journal = {arXiv:1707.06342},
# Year = {2017}
# }
# fixed size: center 224x224 crop from resized image with 256x256
# The accuracy should be: Top-1: 0.6734, Top-5: 0.8792
# If you have any problem, please feel free to contact Jian-Hao Luo (luojh@lamda.nju.edu.cn).
# 2017-07-29
name: "GoogleNet"
layer {
name: "data"
type: "Data"
top: "data"
top: "label"
include {
phase: TRAIN
}
transform_param {
name: prune_llm
channels:
- pytorch
- nvidia
- conda-forge
- defaults
dependencies:
- _libgcc_mutex=0.1=main
- _openmp_mutex=5.1=1_gnu
- blas=1.0=mkl