Skip to content

Latest commit

 

History

History
20 lines (14 loc) · 568 Bytes

README.md

File metadata and controls

20 lines (14 loc) · 568 Bytes

Segformer Demo

How to run demo

  • Use the following command to run the Segformer Encoder model (Classification):

    pytest tests/ttnn/integration_tests/segformer/test_segformer_encoder.py

  • Use the following command to run the Segformer Decoder module model:

    pytest tests/ttnn/integration_tests/segformer/test_segformer_decode_head.py

  • Use the following command to run the Segformer full model (Segmentation):

    pytest tests/ttnn/integration_tests/segformer/test_segformer_for_semantic_segmentation.py