From 8d96ddf4e57d2d3b9d5c91f298b8b0c76690abe0 Mon Sep 17 00:00:00 2001
From: whai362 <wangwenhai362@163.com>
Date: Wed, 30 Jun 2021 15:32:02 +0800
Subject: [PATCH] Update README.md

---
 classification/README.md | 12 ++++++++++++
 1 file changed, 12 insertions(+)

diff --git a/classification/README.md b/classification/README.md
index b3d4e8d..3b03535 100644
--- a/classification/README.md
+++ b/classification/README.md
@@ -108,5 +108,17 @@ To train PVT-Small on ImageNet on a single node with 8 gpus for 300 epochs run:
 sh dist_train.sh configs/pvt/pvt_small.py 8 --data-path /path/to/imagenet
 ```
 
+## Calculating FLOPS & Params
+
+```
+python get_flops.py pvt_v2_b2
+```
+This should give
+```
+Input shape: (3, 224, 224)
+Flops: 4.04 GFLOPs
+Params: 25.36 M
+```
+
 ## License
 This repository is released under the Apache 2.0 license as found in the [LICENSE](LICENSE) file.