search

BiomedSciAI / fuse-med-ml

A python framework accelerating ML based discovery in the medical field by encouraging code reuse. Batteries included :)
Apache License 2.0
134 stars 34 forks source link

ViT #203

Open egozi opened 1 year ago

egozi commented 1 year ago

add a few things:

  1. default parameters for depth, heads etc.
  2. standard models like - vit_base, vit_large, …
  3. the interpolation module for diff image sizes.
  4. What about pre trained weights? and an option flag.

Originally posted by @egozi in https://github.com/BiomedSciAI/fuse-med-ml/issues/200#issuecomment-1301901903