Commits

Commits on Jul 13, 2024

fix fa3 version
guolinke
committedJul 13, 2024

Commits on Jul 12, 2024

update yml
Your Name
committedJul 12, 2024
add fa3 compilation
Your Name
committedJul 12, 2024

Commits on Jul 11, 2024

add device code for adam (#61 )
guolinke
committedJul 11, 2024

Commits on Jul 1, 2024

fix bug in setting total steps from max-epoch (#59 )
guolinke
committedJul 1, 2024

Commits on Jun 26, 2024

Add Volces CR (#56 )
leasunhy
committedJun 26, 2024

Commits on Jun 24, 2024

Fix docker building (#55 )
leasunhy
committedJun 24, 2024

Commits on Jun 23, 2024

fix ema (#54 )
guolinke
committedJun 23, 2024

Commits on Jun 12, 2024

compatible with torch 2.3 (#52 )

robotcator
and
Your Name
committedJun 12, 2024

Commits on Jun 8, 2024

fix dummy batch (#51 )
guolinke
committedJun 8, 2024

Commits on Jun 7, 2024

cuda 12.4 (#50 )

robotcator
and
Your Name
committedJun 7, 2024

Commits on May 9, 2024

update requirement (#35 )
PKUfjh
committedMay 9, 2024
add sm90 compiling flag (#44 )
robotcator
committedMay 9, 2024

Commits on Aug 28, 2023

fix ema (#33 )

robotcator
and
jixh
committedAug 28, 2023

Commits on Jun 12, 2023

Update setup.py
guolinke
committedJun 12, 2023

Commits on Jun 9, 2023

fix import error
guolinke
committedJun 9, 2023

Commits on Jun 7, 2023

support the team name in wandb projct
guolinke
committedJun 7, 2023
upgrade to torch 2.0.1
guolinke
committedJun 7, 2023
Update docker_rdma_latest.yml
guolinke
committedJun 7, 2023
remove old dockerfile
guolinke
committedJun 7, 2023
support pytorch 2.0 (#30 )

committedJun 7, 2023

Commits on Jun 2, 2023

fix docker build error
guolinke
committedJun 2, 2023
by default install wandb
guolinke
committedJun 2, 2023
support Wandb (#29 )
guolinke
committedJun 2, 2023

Commits on Apr 19, 2023

fix max size in adam (#27 )
guolinke
committedApr 19, 2023

Commits on Apr 15, 2023

allow --keep-last-epochs == 0 (#26 )

ZhouGengmo
and
zhougm
committedApr 15, 2023

Commits on Mar 31, 2023

set default --required-batch-size-multiple to 1
guolinke
committedMar 31, 2023

Commits on Mar 29, 2023

fix loading model logs (#24 )
ShuqiLu
committedMar 29, 2023

Commits on Mar 28, 2023

Add new wheels (#23 )
robotcator
committedMar 28, 2023

Commits on Mar 10, 2023

add --disable_extension flags to disable cuda extension (#21 )
robotcator
committedMar 10, 2023

Commits on Mar 9, 2023

pass arg to switch ckp on/off (#22 )
ZiyaoLi
committedMar 9, 2023

Commits on Mar 5, 2023

fix numpy seed key ignorance (#20 )
ZiyaoLi
committedMar 5, 2023
support layernorm with dim 192
guolinke
committedMar 5, 2023

Commits on Feb 25, 2023

install bc by default
guolinke
committedFeb 25, 2023

Commits on Feb 23, 2023

filter params in ema (#19 )
teslacool
committedFeb 23, 2023