LOG-postech / SasshaLinks
Official Pytorch code for "SASSHA: Sharpness-aware Adaptive Second-order Optimization With Stable Hessian Approximation"
β14Updated 3 months ago
Alternatives and similar repositories for Sassha
Users that are interested in Sassha are comparing it to the libraries listed below
Sorting:
- Code for UAI 2025 paper "Critical Influence of Overparameterization on Sharpness-aware Minimization"β19Updated 2 months ago
- π¨ Malet (Machine Learning Experiment Tool) is a tool for efficient machine learning experiment execution, logging, analysis, and plot maβ¦β17Updated 2 months ago
- β14Updated 4 months ago
- β27Updated 4 months ago
- β25Updated last year
- β17Updated 2 years ago
- PyHessian is a Pytorch library for second-order based analysis and training of Neural Networksβ746Updated last week
- [NeurIPS 2021] code for "Taxonomizing local versus global structure in neural network loss landscapes" https://arxiv.org/abs/2107.11228β19Updated 3 years ago
- PyTorch implementation of Mixer-nano (#parameters is 0.67M, originally Mixer-S/16 has 18M) with 90.83 % acc. on CIFAR-10. Training from sβ¦β33Updated 3 years ago
- Comparison of method "Pruning at initialization prior to training" (Synflow/SNIP/GraSP) in PyTorchβ16Updated last year
- [KDD 2023] code for "Test accuracy vs. generalization gap: model selection in NLP without accessing training or testing data" https://arxβ¦β12Updated 2 years ago
- Pytorch implementation of the paper "SNIP: Single-shot Network Pruning based on Connection Sensitivity" by Lee et al.β108Updated 6 years ago
- tiny-imagenet dataset downloader & reader using tensorflow_datasets (tfds) apiβ20Updated 5 years ago
- Source code of "Task arithmetic in the tangent space: Improved editing of pre-trained models".β102Updated 2 years ago
- Pretrained models on CIFAR10/100 in PyTorchβ359Updated 2 months ago
- Implement of Dynamic Model Pruning with Feedback with pytorchβ40Updated 3 years ago
- β191Updated 4 years ago
- PyTorch implementation for Vision Transformer[Dosovitskiy, A.(ICLR'21)] modified to obtain over 90% accuracy FROM SCRATCH on CIFAR-10 witβ¦β198Updated last year
- β226Updated 11 months ago
- Influence Functions with (Eigenvalue-corrected) Kronecker-Factored Approximate Curvatureβ156Updated 3 weeks ago
- Official Implementation of LANTERN (ICLR'25) and LANTERN++(ICLRW-SCOPE'25)β15Updated 4 months ago
- β36Updated 2 years ago
- This repository contains a Pytorch implementation of the paper "The Lottery Ticket Hypothesis: Finding Sparse, Trainable Neural Networks"β¦β332Updated last year
- pytorch-tiny-imagenetβ179Updated last year
- Efficient PyTorch Hessian eigendecomposition tools!β374Updated last year
- Pretrained TorchVision models on CIFAR10 dataset (with weights)β676Updated 2 years ago
- A curated list of papers of interesting empirical study and insight on deep learning. Continually updating...β332Updated last week
- β70Updated 7 months ago
- β34Updated last year
- [Deep Unlearning-PyTorch] Class Forgetting as in paper "Deep Unlearning: Fast and Efficient Training-free Approach to Controlled Forgettiβ¦β15Updated 11 months ago