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 |