Daniel Suess 986de90360

Register orm_act layers as leaf modules

2023-08-10 15:37:26 +10:00

..

data

Replace deprecated NumPy aliases of builtin types

2023-07-03 22:24:25 +03:00

layers

Fix missing norm call in Mlp forward (not used by default, but can be enabled for normformer MLP scale). Fix #1851 fix #1852

2023-08-03 11:36:30 -07:00

loss

Remove inplace operators when calculating the loss

2022-10-17 10:41:21 +02:00

models

Register orm_act layers as leaf modules

2023-08-10 15:37:26 +10:00

optim

import lion in __init__.py

2023-07-27 09:26:57 +08:00

scheduler

Scheduler update, add v2 factory method, support scheduling on updates instead of just epochs. Add LR to summary csv. Add lr_base scaling calculations to train script. Fix #1168

2022-10-07 10:43:04 -07:00

utils

clip gradients with update

2023-04-19 23:36:20 -07:00

__init__.py

Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models

2022-12-06 15:00:06 -08:00

version.py

Version 0.9.5 prep for release. README update

2023-08-03 15:16:46 -07:00