Giters

HuYang719 / DistributedViT

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Distributed ViT

Build

cd ./distributedViT
make

Test

  1. First change the image filepath in test_transformer() in transformer.c
  2. Run with command:
./darknet transformer test ./cfg/vit-test.cfg

or test the inference time

time ./darknet transformer test ./cfg/vit-test.cfg

About

Languages

Language:C 69.0%Language:Cuda 15.2%Language:C++ 14.8%Language:CMake 0.3%Language:Makefile 0.3%Language:Python 0.2%Language:Shell 0.1%