Ross Wightman b8c8550841 Data improvements. Improve train support for in_chans != 3. Add wds dataset support from bits_and_tpu branch w/ fixes and tweaks. TFDS tweaks.		2022-09-29 16:42:58 -07:00
..
data	Data improvements. Improve train support for in_chans != 3. Add wds dataset support from bits_and_tpu branch w/ fixes and tweaks. TFDS tweaks.	2022-09-29 16:42:58 -07:00
loss	…
models	Add brute-force checkpoint remapping option	2022-09-23 16:07:03 -07:00
optim	Add Adan optimizer	2022-09-23 16:05:52 -07:00
scheduler	fix: change milestones to decay-milestones	2022-05-10 07:57:19 +09:00
utils	Refactor device handling in scripts, distributed init to be less 'cuda' centric. More device args passed through where needed.	2022-09-23 16:08:59 -07:00
__init__.py	…
version.py	version 0.6.11	2022-09-23 13:54:56 -07:00