This commit is contained in:
Yao Wang 2024-01-22 21:02:16 +08:00
parent 58fa893d4d
commit c8f2babd76
8 changed files with 300 additions and 7 deletions

1
Code/evaluation.sh Executable file
View file

@ -0,0 +1 @@
python evaluation.py --model 'bert' --ckpt './VisSalFormer_weights.tar' --device 'cuda'