Skip to content
GitLab
Explore
Sign in
Tags
Tags give the ability to mark specific points in history as being important
This project is mirrored from
https://github.com/lucidrains/pytorch-custom-utils
. Pull mirroring updated
Sep 18, 2024
.
0.0.21
a452e954
·
0.0.21
·
Jul 26, 2024
0.0.20
7bcbc9be
·
Update setup.py
·
Jun 07, 2024
0.0.19
22927efb
·
0.0.19
·
Jun 05, 2024
0.0.18
b94eaad1
·
few missing wraps
·
Feb 02, 2024
0.0.17
8500c972
·
another commonly used masking fn
·
Jan 28, 2024
0.0.16
9f7fbc32
·
fix masked mean
·
Jan 27, 2024
0.0.15
76e9a2c6
·
add masked mean
·
Jan 26, 2024
0.0.14
c0649fc6
·
add masked mean
·
Jan 26, 2024
0.0.12
c7483d73
·
decorator for getting total parameters of a module
·
Jan 13, 2024
0.0.11
6092e18b
·
helper function for grad accum and accelerator.no_sync context
·
Jan 11, 2024
0.0.10
85ee52c9
·
fix warmup period of 0 for optimizer with warmup
·
Jan 11, 2024
0.0.9
3367cb8d
·
decorator that automatically unwraps a model with accelerator instance in a trainer class
·
Jan 07, 2024
0.0.8
74ccf29d
·
use optree
·
Dec 22, 2023
0.0.7
065d07e7
·
add another decorator that autocasts inputs into a method onto the device for...
·
Dec 22, 2023
0.0.6
58652c70
·
handy context manager for running a lot of wandb experiments with a trainer...
·
Dec 21, 2023
0.0.5
8ac6aed4
·
bundle warmup, scheduler, gradient clipping and optimizer into one class
·
Dec 18, 2023
0.0.4
b5b41f3e
·
simple fn for getting adam optimizer with maybe weight decay, and omits gammas...
·
Dec 18, 2023
0.0.3
8440680c
·
quick device access
·
Dec 15, 2023
0.0.2
371b14bc
·
quick device access
·
Dec 15, 2023
0.0.1
6b08bd93
·
start drying some software engineering stuff outside of the actual research
·
Dec 15, 2023