evonneng
4801a99b2a
improve memory consumption by clearing cache + move image to cpu ( #412 )
...
* reduced mem on train; figuring out memory leak on visualizer
* steadily below 10G memory
* cpu loading image
* update docstrings
* added scrolling
* Revert "update docstrings"
This reverts commit a914387cf8177c3dd6f1f37351ee7f
2022-09-12 23:00:44 -07:00
evonneng
d50c71ac43
update docstrings and random cleanups ( #387 )
...
* fixing docstrings and small nits; up to fileds
* finished adding and fixing docstrings
2022-09-10 23:09:44 -07:00
Ethan Weber
512a51f714
Fixed friends dataset w/o caching ( #365 )
...
* fixing friends dataloading and nerfw running
* put dataset back to normal
2022-09-07 12:24:29 -07:00
evonneng
56113e6c75
[WIP] dataloading cleanup ( #332 )
...
* testing; need to merge
* fixing wrong base file in models/base
* pushed input parser into datasets class to abstract away fetching data in dataloader
* slight name chage and all
* nits
* added dataparser
* added licnese
2022-09-06 13:54:35 -07:00
evonneng
Ethan Weber