* edgenext refactored for torchscript compat, stage base organization * slight refactor of ConvNeXt to match some EdgeNeXt additions * remove use of funky LayerNorm layer in ConvNeXt and just use nn.LayerNorm and LayerNorm2d (permute) |
||
---|---|---|
.. | ||
data | ||
loss | ||
models | ||
optim | ||
scheduler | ||
utils | ||
__init__.py | ||
version.py |