ppo

2025-08-14 17:41:37 +08:00 · 2020-10-21 14:26:37 +05:30
parent cd0b7eee10
commit 933bc0cbf2
7 changed files with 704 additions and 2 deletions
--- a/setup.py
+++ b/setup.py
@ -19,7 +19,9 @@ setuptools.setup(
                                               'test.*')),
    install_requires=['labml>=0.4.41',
                      'labml_helpers',
-                      'torch'],
+                      'torch',
+                      'einops',
+                      'numpy'],
    classifiers=[
        "Programming Language :: Python :: 3",
        "License :: OSI Approved :: MIT License",