11 results found Sort:

cszn
KAIR cszn
642
3.0k
mit
47
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
419 commits to master branch, last one 2 months ago
PistonY
torch-toolbox PistonY
55
417
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 7 months ago
1adrianb
pytorch-estimate-flops 1adrianb
22
305
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
22
210
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
67 commits to main branch, last one 20 days ago
amirgholami
ai_and_memory_wall amirgholami
25
208
mit
3
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 9 months ago
ProjectPhysX
OpenCL-Benchmark ProjectPhysX
23
174
other
7
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
39 commits to master branch, last one 3 days ago
vra
flopth vra
9
123
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
88 commits to master branch, last one about a month ago
LukasHedegaard
pytorch-benchmark LukasHedegaard
9
93
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 about a year ago
gmalivenko
onnx-opcounter gmalivenko
21
88
apache-2.0
2
Count number of parameters / MACs / FLOPS for ONNX models.
cli macs onnx flops onnx-models machine-learning
Created 2020-12-17
20 commits to master branch, last one 2 months ago
tokusumi
keras-flops tokusumi
8
51
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 4 years ago
kyo-takano
chinchilla kyo-takano
4
43
apache-2.0
1
A toolkit for scaling law research ⚖
llm flops pretraining scaling-laws deep-learning reinforcement-learning
Created 2024-01-31
17 commits to master branch, last one 15 days ago