GitHub / rasbt / pytorch-memory-optim
This code repository contains the code used for my "Optimizing Memory Usage for Training LLMs and Vision Transformers in PyTorch" blog post.
JSON API: http://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/rasbt%2Fpytorch-memory-optim
PURL: pkg:github/rasbt/pytorch-memory-optim
Stars: 91
Forks: 10
Open issues: 0
License: apache-2.0
Language: Python
Size: 1.14 MB
Dependencies parsed at: Pending
Created at: almost 2 years ago
Updated at: 2 months ago
Pushed at: almost 2 years ago
Last synced at: 2 months ago
Topics: deep-learning, llm, memory-optimization, pytorch, vision, vit
Loading...