You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

Ross Wightman 17cdee7354 Fix C&P patch_size error, and order of op patch_size arg resolution bug. Remove a test vit model.		4 years ago
..
data	Add epoch-repeats arg to multiply the number of dataset passes per epoch. Currently for iterable datasets (read TFDS wrapper) only.	4 years ago
loss	asymmetric_loss	4 years ago
models	Fix C&P patch_size error, and order of op patch_size arg resolution bug. Remove a test vit model.	4 years ago
optim	Add benchmark.py script, and update optimizer factory to be more friendly to use outside of argparse interface.	4 years ago
scheduler	Merge branch 'logger' into features. Change 'logger' to '_logger'.	4 years ago
utils	Initial AGC impl. Still testing.	4 years ago
__init__.py	Monster commit, activation refactor, VoVNet, norm_act improvements, more	4 years ago
version.py	Update README for AGC and bump version to 0.4.4	4 years ago