Update README.md

tejas1904 · Mar 30, 2023 · bbdbedf · bbdbedf
1 parent 00a2609
commit bbdbedf
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/README.md b/README.md
@@ -34,7 +34,7 @@ Then import model
 import tensorflow as tf
 from models import SegFormer_B3
 model = SegFormer_B3(input_shape = (224, 224, 3), num_classes = 19)
-print(model.summary(expand_nested=True))
+print(model.summary())
 ```