Ruslan Baikulov
|
158bf129c4
|
Replace deprecated NumPy aliases of builtin types
|
2023-07-03 22:24:25 +03:00 |
Ross Wightman
|
fd592ec86c
|
Fix an issue with FastCollateMixup still using device
|
2023-05-10 08:55:38 -07:00 |
Junming Chen
|
569d114ef7
|
Fix device problem
Before, the one_hot could only run in device='cuda'. Now it will run on input device automatically.
|
2022-04-19 11:53:18 +08:00 |
|
Ross Wightman
|
656757d26b
|
Fix MobileNetV2 head conv size for multiplier < 1.0. Add some missing modification copyrights, fix starting date of some old ones.
|
2022-01-14 16:28:27 -08:00 |
masafumi
|
047a5ec05f
|
Fix bugs that Mixup does not work device=cpu
|
2021-10-12 23:51:46 +09:00 |
|
Ross Wightman
|
47a7b3b5b1
|
More flexible mixup mode, add 'half' mode.
|
2020-09-07 20:03:06 -07:00 |
|
Ross Wightman
|
8c9814e3f5
|
Final cleanup of mixup/cutmix. Element/batch modes working with both collate (prefetcher active) and without prefetcher.
|
2020-08-12 17:01:32 -07:00 |
|
Ross Wightman
|
cd23f55397
|
Fix mixed prec issues with new mixup code
|
2020-08-11 12:17:43 -07:00 |
|
Ross Wightman
|
f471c17c9d
|
More cutmix/mixup overhaul, ready to kick-off some trials.
|
2020-08-11 00:10:33 -07:00 |
|
Ross Wightman
|
92f2d0d65d
|
Merge branch 'master' into cutmix. Fixup a few issues.
|
2020-08-07 15:59:52 -07:00 |
|
Ross Wightman
|
6c17d57a2c
|
Fix some attributions, add copyrights to some file docstrings
|
2020-07-27 13:44:56 -07:00 |
|
Ross Wightman
|
670c61b28f
|
Some cutmix/mixup cleanup/fixes
|
2020-02-17 11:00:54 -08:00 |
|
Ross Wightman
|
b3cb5f3275
|
Working on CutMix impl as per #8, integrating with Mixup, currently experimenting...
|
2020-02-16 20:09:11 -08:00 |
|
Ross Wightman
|
232ab7fb12
|
Working on an implementation of AugMix with JensenShannonDivergence loss that's compatible with my AutoAugment and RandAugment impl
|
2019-12-20 23:04:11 -08:00 |
|
Ross Wightman
|
aa4354f466
|
Big re-org, working towards making pip/module as 'timm'
|
2019-06-19 17:20:51 -07:00 |