* keep most of net in BCHW layout, performance appears same, can remove static resolution attribs and features easier to use * add F.sdpa, decent gains in pt 2.1 * tweak crop pct based on eval |
||
---|---|---|
.. | ||
data | ||
layers | ||
loss | ||
models | ||
optim | ||
scheduler | ||
utils | ||
__init__.py | ||
version.py |