Skip to content

Latest commit

 

History

History
16 lines (11 loc) · 564 Bytes

readme.md

File metadata and controls

16 lines (11 loc) · 564 Bytes

vit-malware-detector

refined vit train scripts for malware detection to test with risc-v cdac accelerator

changes

  • removed deprecated tfa (tensorflow_addons) in favor of keras
  • added requirements file for easy setup with python venv
  • integrated tensorflow-metal for faster local tests in macbook

todo

  • make sense of the code
  • host the trained model on huggingface to do inferences in google colab easily
  • more detailed output in terminal and in diagrams for diagonosis
  • make the script platform independant and human friendly