Граф коммитов

2633 Коммитов

Автор SHA1 Сообщение Дата
Shital Shah aa2f8fafca added mixed prec enabled flag 2020-05-18 03:12:42 -07:00
Shital Shah 8969661426 fix getting dist rank 2020-05-18 03:12:42 -07:00
Shital Shah 00a5d5f16b aded disitributed metrics, reduce with different ops 2020-05-18 03:12:42 -07:00
Shital Shah 00646b8e41 slighly cleaner way to create torch device 2020-05-18 03:12:42 -07:00
Shital Shah b4c4cd506d minor fixes for the toy run 2020-05-18 03:12:42 -07:00
Shital Shah 63b3ec140d separate out losses in module, cuda synchronize, proper warmup handling, default amp at O2, add BN weight decay, nvidia benchmark config, 2020-05-18 03:12:42 -07:00
Shital Shah 1cdcb8d7ba set rank refactor, add local_rank to seed 2020-05-18 03:12:42 -07:00
Shital Shah ee99587b14 distributed and mixed precision enablement 2020-05-18 03:12:42 -07:00
Shital Shah cec9c16780 imagenet support for handcrafted models 2020-05-18 03:12:42 -07:00
Shital Shah 8a104c53aa added salsa optomizer 2020-05-18 03:12:42 -07:00
Shital Shah c3e1eff660 increase max batch for toy where val is needed 2020-05-18 03:12:42 -07:00
Shital Shah 0a5d3cf8ac ImageNet toy mode cell outs verified, better readibility in macro builder, move aux tower stride to yaml, prev reduction based on model stems to support imagenet, remove adaptive avg pool from aux tower, remove bn from pool op, converted original darts genotype to archai format 2020-05-18 03:12:42 -07:00
Shital Shah 6720187057 Added ImageNet toy run, multi_step LRs, separate out imagenet folder class, add hue in imagenet aug, module name guess for pre-crafted models, add reduction property in stems, remove imagenet from stem/pool names, combine s4 stem in s2 stem, tune toy.yaml for more realistic toy mode 2020-05-18 03:12:42 -07:00
Shital Shah 76467a1168 log main training params 2020-05-18 03:12:42 -07:00
Shital Shah dbaedac6bb allow faux search in manual mode, remove aux_tower redundent flag 2020-05-18 03:12:42 -07:00
Shital Shah de6a0e2685 added manual in all toy run 2020-05-18 03:12:41 -07:00
Shital Shah e40777ef35 resnet eval working in toy 2020-05-18 03:12:41 -07:00
Shital Shah d4d97818d2 enable manual models 2020-05-18 03:12:41 -07:00
Shital Shah bdc37f161c detail and summary reports 2020-05-18 03:12:41 -07:00
Shital Shah b9127a995f added summary 2020-05-18 03:12:41 -07:00
Shital Shah 79948a3786 experiment report basic 2020-05-18 03:12:41 -07:00
Shital Shah cf01cb7777 image state fixes, report gen initial 2020-05-18 03:12:41 -07:00
Shital Shah 30a57ca660 added sample models 2020-05-18 03:12:41 -07:00
Shital Shah 29601603db remove gs params from darts 2020-05-18 03:12:41 -07:00
Shital Shah a77b7bafa8 proper log resets, computer imagenet stats 2020-05-18 03:12:41 -07:00
Shital Shah a87682bd4f added comment for zero op 2020-05-18 03:12:41 -07:00
Shital Shah 06969a8e00 new state management for model desc 2020-05-18 03:12:41 -07:00
Shital Shah 6c44c6e65c added model vis script 2020-05-18 03:12:41 -07:00
Shital Shah 57f9cac230 log main model desc params 2020-05-18 03:12:41 -07:00
Shital Shah cd6a01a5f4 added eval only test 2020-05-18 03:12:41 -07:00
Shital Shah 0f9f83d59a fix log message, use trainer title in log, 2020-05-18 03:12:41 -07:00
Shital Shah c3e97d378b support general paths in yaml 2020-05-18 03:12:41 -07:00
Shital Shah af1d639c6e initial 2020-05-18 03:11:07 -07:00