Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/OliverRensu/ARM
/ types & classes
Types & classes
24 in github.com/OliverRensu/ARM
⨍
Functions
161
◇
Types & classes
24
↓ 3 callers
Class
ARM
Finetuning/models_mamba.py:203
↓ 3 callers
Class
VisionMamba
models_pretrain.py:285
↓ 2 callers
Class
SmoothedValue
Track a series of values and provide access to smoothed values over a window or the global series average.
utils/misc.py:19
↓ 2 callers
Class
SmoothedValue
Track a series of values and provide access to smoothed values over a window or the global series average.
Finetuning/util/misc.py:24
↓ 1 callers
Class
Block
models_pretrain.py:164
↓ 1 callers
Class
Block
Finetuning/models_mamba.py:82
↓ 1 callers
Class
CrossAttention
models_pretrain.py:55
↓ 1 callers
Class
DecoderBlock
models_pretrain.py:86
↓ 1 callers
Class
Mlp
models_pretrain.py:35
↓ 1 callers
Class
PatchEmbed
2D Image to Patch Embedding
models_pretrain.py:108
↓ 1 callers
Class
PatchEmbed
2D Image to Patch Embedding
Finetuning/models_mamba.py:28
↓ 1 callers
Class
SwiGLU
models_pretrain.py:137
↓ 1 callers
Class
SwiGLU
Finetuning/models_mamba.py:55
Class
Block
mamba_simple.py:541
Class
Block
Finetuning/mamba_simple.py:544
Class
LARS
LARS optimizer, no rate scaling or weight decay for parameters <= 1D.
Finetuning/util/lars.py:14
Class
Mamba
mamba_simple.py:35
Class
Mamba
Finetuning/mamba_simple.py:36
Class
MetricLogger
utils/misc.py:81
Class
MetricLogger
Finetuning/util/misc.py:86
Class
NativeScalerWithGradNormCount
utils/misc.py:236
Class
NativeScalerWithGradNormCount
Finetuning/util/misc.py:251
Class
RandomResizedCrop
RandomResizedCrop for matching TF/TPU implementation: no for-loop is used. This may lead to results different with torchvision's version.
utils/crop.py:16
Class
RandomResizedCrop
RandomResizedCrop for matching TF/TPU implementation: no for-loop is used. This may lead to results different with torchvision's version.
Finetuning/util/crop.py:15