alan-turing-institute / affinity-vae

Self-supervised method for disentanglement, clustering and classification of objects in multidimensional image data
BSD 3-Clause "New" or "Revised" License
13 stars 7 forks source link

Updated toml file for different systems #271

Closed crangelsmith closed 9 months ago

crangelsmith commented 10 months ago

Installation are different depending what are we trying to do:

Suggestion:

  1. python -m pip install . should have a minimal installation needed for everything, but it doesn't have all dependencies.
  2. python -m pip install ."[baskerville]" will have the needed libraries for running in baskervile (which need to be complemented with torch and tochvision modules).
  3. python -m pip install ."[all]" should have everything to train the model locally.
  4. python -m pip install ."[napari]" with have the base version of dependencies from 1. plus the version of pydantic that is compatible with napari.
  5. python -m pip install ."[tests]" should have everything to train the model locally and run tests.