From 2a31ea2f3dccae87aa1c0d7adc12849fdcd3699f Mon Sep 17 00:00:00 2001 From: Gongfan Fang Date: Sun, 17 Nov 2024 17:33:28 +0800 Subject: [PATCH] Update README.md --- README.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/README.md b/README.md index ecedb98..2a1b217 100644 --- a/README.md +++ b/README.md @@ -192,7 +192,7 @@ pruner = tp.pruner.MetaPruner( # We can always choose MetaPruner if sparse train pruning_ratio=0.5, # remove 50% channels, ResNet18 = {64, 128, 256, 512} => ResNet18_Half = {32, 64, 128, 256} # pruning_ratio_dict = {model.conv1: 0.2, model.layer2: 0.8}, # customized pruning ratios for layers or blocks ignored_layers=ignored_layers, - round_to=8, # It's recommended to round dim/channels to 4x or 8x for acceleration. Please see: https://docs.nvidia.com/deeplearning/performance/dl-performance-convolutional/index.html + round_to=8, # It's recommended to round dims/channels to 4x or 8x for acceleration. Please see: https://docs.nvidia.com/deeplearning/performance/dl-performance-convolutional/index.html ) # 3. Prune & finetune the model