11 results found Sort:

cszn
KAIR cszn
653
3.1k
mit
46
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 5 months ago
PistonY
torch-toolbox PistonY
55
418
bsd-3-clause
11
🛠 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 10 months ago
1adrianb
pytorch-estimate-flops 1adrianb
22
303
bsd-3-clause
8
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
218
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
68 commits to main branch, last one 2 months ago
amirgholami
ai_and_memory_wall amirgholami
26
213
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 about a year ago
ProjectPhysX
OpenCL-Benchmark ProjectPhysX
26
193
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
51 commits to master branch, last one 4 days ago
vra
flopth vra
9
129
mit
2
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 4 months ago
LukasHedegaard
pytorch-benchmark LukasHedegaard
11
98
apache-2.0
3
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
22
89
apache-2.0
3
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 4 months ago
tokusumi
keras-flops tokusumi
8
54
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
48
apache-2.0
1
A toolkit for scaling law research âš–
llm flops pretraining scaling-laws deep-learning reinforcement-learning
Created 2024-01-31
23 commits to master branch, last one about a month ago