tanda
tanda copied to clipboard
Make loading pre-trained TAN a one-liner
Right now we have to load in:
- The TAN checkpoint
- Config file
- TFs
- dims
(4) should be merged into (2); (3) should also be merged into (2) (see e.g. scuba repo); and then everything should be put into one folder so everything can be auto-loaded with one path
Just to clarify from our discussion: you should be able to pass in either a TAN object or a string as the first argument to TANDAImageDataGenerator