openpi / scripts /train_pytorch.py

Commit History

Fix PyTorch assets save path
b29f81b

Kevin Black commited on

Fixes for smooth eval
2502a4b

Kevin Black commited on

Fix lint errors
8999e55

Kevin Black commited on

remove ema, remove redundant compute for data loading, memory opt for float32, and set float32 by default for finetuning
aa7853c

klldmofashi commited on

add pytorch support for openpi
5546575

klldmofashi commited on