early training loss as expected

2024-03-11 13:34:04 +00:00
parent fab2b3240b
commit 2a01487494
5 changed files with 232 additions and 8 deletions
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -48,6 +48,8 @@ opencv-python = "^4.9.0.80"
 diffusers = "^0.26.3"
 torchvision = "^0.17.1"
 h5py = "^3.10.0"
+robomimic = "0.2.0"
+timm = "^0.9.16"


 [tool.poetry.group.dev.dependencies]