7 Commits

Author	SHA1	Message	Date
Ross Wightman	e2b8d44ff0	Halo, bottleneck attn, lambda layer additions and cleanup along w/ experimental model defs ... * align interfaces of halo, bottleneck attn and lambda layer * add qk_ratio to all of above, control q/k dim relative to output dim * add experimental haloregnetz, and trionet (lambda + halo + bottle) models	2021-10-06 16:32:48 -07:00
Ross Wightman	b1c2e3eb92	Match rel_pos_indices attr rename in conv branch	2021-09-30 23:19:05 -07:00
Ross Wightman	b49630a138	Add relative pos embed option to LambdaLayer, fix last transpose/reshape.	2021-09-30 22:45:09 -07:00
Ross Wightman	5bd04714e4	Cleanup weight init for byob/byoanet and related	2021-09-05 15:34:05 -07:00
Ross Wightman	0721559511	Improved (hopefully) init for SA/SA-like layers used in ByoaNets	2021-05-04 21:40:39 -07:00
Ross Wightman	e15c3886ba	Defaul lambda r=7. Define '26t' stage 4/5 256x256 variants for all of bot/halo/lambda nets for experiment. Add resnet50t for exp. Fix a few comments.	2021-04-29 10:58:49 -07:00
Ross Wightman	ce62f96d4d	ByoaNet with bottleneck transformer, lambda resnet, and halo net experiments	2021-04-12 09:38:02 -07:00