Commit Graph

5 Commits

Author SHA1 Message Date
Carlos Chavez 80e2422967
sync trainer with main branch 2022-11-16 10:39:20 -05:00
Carlos Chavez d600078008
Add options and local inference
Added options to:
- Disable Inference (it consumes about 2gb of VRAM even when not active)
- Disable wandb

and:
- if no hftoken is provided it just fills it with nothing so it doesn't argues
- if wandb is not enabled, save the inference outputs to a local folder along with information about it
2022-11-14 22:08:16 -05:00
lopho cd0910e82d
Parse booleans in argument parser
true, yes or 1 correspond to True, else False.
2022-11-12 11:14:48 +01:00
Anthony Mercurio 624f0f14af
correctly set args 2022-11-11 08:25:27 -07:00
harubaru 1f5b671b67 relicense 2022-11-10 12:59:53 -07:00