First real commit, simxarm env added with torchrl!

This commit is contained in:
Cadene
2024-01-29 12:49:30 +00:00
parent 0396980450
commit 1144819c29
11 changed files with 437 additions and 1 deletions

View File

@@ -0,0 +1,21 @@
import os
import zipfile
import gdown
def download():
url = "https://drive.google.com/uc?id=1nhxpykGtPDhmQKm-_B8zBSywVRdgeVya"
download_path = "data.zip"
gdown.download(url, download_path, quiet=False)
print("Extracting...")
with zipfile.ZipFile(download_path, "r") as zip_f:
for member in zip_f.namelist():
if member.startswith("data/xarm") and member.endswith(".pkl"):
print(member)
zip_f.extract(member=member)
os.remove(download_path)
if __name__ == "__main__":
download()