Jacob Berg issues

Repositories
Issues
Comments

Results 1 issues of


                                            Jacob Berg

Robocasa Language Embedding Cuda Out of Memory Error

In line 222 of the Robocasa branch of [robomimic/utils/train_utils.py](https://github.com/ARISE-Initiative/robomimic/blob/9a4b448cd4eb1d1a4a8f97e2c1c51717782b397f/robomimic/utils/train_utils.py#L220), upon dataset creation, the dataset kwargs are deepcoppied. Since the language embedding model is one of the dataset_kwargs, this makes a...