Ross Wightman 05313940e2 Add cs3darknet_x, cs3sedarknet_l, and darknetaa53 weights from TPU sessions. Move SE btwn conv1 & conv2 in DarkBlock. Improve SE/attn handling in Csp/DarkNet. Fix leaky_relu bug on older csp models.

2022-07-15 16:55:16 -07:00

..

data

Change set -> dict w/ None keys for dataset split synonym search, so always consistent if more than 1 exists. Fix #1224

2022-07-07 15:33:53 -07:00

loss

…

models

Add cs3darknet_x, cs3sedarknet_l, and darknetaa53 weights from TPU sessions. Move SE btwn conv1 & conv2 in DarkBlock. Improve SE/attn handling in Csp/DarkNet. Fix leaky_relu bug on older csp models.

2022-07-15 16:55:16 -07:00

optim

Fix bug introduced in non layer_decay weight_decay application. Remove debug print, fix arg desc.

2022-02-28 17:06:32 -08:00

scheduler

fix: change milestones to decay-milestones

2022-05-10 07:57:19 +09:00

utils

disable nvfuser for jit te/legacy modes (for PT 1.12+)

2022-07-13 10:34:34 -07:00

__init__.py

…

version.py

version 0.6.5 for pypi release

2022-07-10 16:43:23 -07:00