11 results found Sort:

cszn
KAIR cszn
606
2.8k
mit
45
Image Restoration Toolbox (PyTorch). Training and testing codes for DPIR, USRNet, DnCNN, FFDNet, SRMD, DPSR, BSRGAN, SwinIR
dpsr sisr srmd dncnn flops bsrgan esrgan ffdnet swinir usrnet pytorch toolbox denoising deep-learning super-resolution image-restoration
Created 2019-12-15
415 commits to master branch, last one about a month ago
PistonY
torch-toolbox PistonY
56
415
bsd-3-clause
12
🛠 Toolbox to extend PyTorch functionalities
cv2 lmdb flops mixup swish cutout arcloss cosloss metrics pytorch toolbox l2softmax lookahead teansform switchnorm autoaugment initializers lmdb-dataset cosinewarmuplr labelsmoothing
Created 2019-05-06
352 commits to master branch, last one 18 days ago
1adrianb
pytorch-estimate-flops 1adrianb
23
301
bsd-3-clause
9
Estimate/count FLOPS for a given neural network using pytorch
flops pytorch deep-learning pytorch-estimate-flops convolutional-neural-networks
Created 2019-02-08
64 commits to master branch, last one 2 years ago
frgfm
torch-scan frgfm
21
204
apache-2.0
7
Seamless analysis of your PyTorch models (RAM usage, FLOPs, MACs, receptive field, etc.)
flops keras python pytorch summary benchmark deep-learning flops-counter pytorch-utils receptive-field deep-neural-networks
Created 2020-03-16
63 commits to main branch, last one 4 months ago
amirgholami
ai_and_memory_wall amirgholami
24
192
mit
2
AI and Memory Wall
gpt flops gpt-2 gpt-3 albert electra bertflops bert-flops bert-model nlp-models memory-wall transformer electra-model training-flops inference-flops albert-transformer
Created 2021-03-15
32 commits to main branch, last one 2 months ago
ProjectPhysX
OpenCL-Benchmark ProjectPhysX
13
124
other
5
A small OpenCL benchmark program to measure peak GPU/CPU performance.
gpu hpc tool flops gpgpu tools opencl bandwidth benchmark benchmarking gpu-computing high-performance-computing
Created 2023-04-30
24 commits to master branch, last one a day ago
vra
flopth vra
9
114
mit
3
A simple program to calculate and visualize the FLOPs and Parameters of Pytorch models, with handy CLI and easy-to-use Python API.
cnn pip flop pypi flops python python3 pytorch deeplearning deep-learning visualization
Created 2019-02-21
85 commits to master branch, last one 5 months ago
LukasHedegaard
pytorch-benchmark LukasHedegaard
9
81
apache-2.0
4
Easily benchmark PyTorch model FLOPs, latency, throughput, allocated gpu memory and energy consumption
gpu flops jetson python pytorch benchmark deep-learning timing-analysis
Created 2022-02-09
73 commits to main branch, last one 9 months ago
gmalivenko
onnx-opcounter gmalivenko
20
79
apache-2.0
1
Count number of parameters / MACs / FLOPS for ONNX models.
cli macs onnx flops onnx-models machine-learning
Created 2020-12-17
10 commits to master branch, last one 3 years ago
tokusumi
keras-flops tokusumi
8
49
mit
1
FLOPs calculator with tf.profiler for neural network architecture written in tensorflow 2.2+ (tf.keras)
flops profiler tensorflow2
Created 2020-08-12
15 commits to master branch, last one 3 years ago
kyo-takano
chinchilla kyo-takano
3
35
apache-2.0
1
A toolkit for scaling law research ⚖
llm flops pretraining scaling-laws deep-learning reinforcement-learning
Created 2024-01-31
16 commits to master branch, last one 2 months ago