Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Enhanced with New Checkpoints, Apple Silicon Support, and Improved Logging #1

Open
wants to merge 6 commits into
base: master
Choose a base branch
from

Conversation

AravindXD
Copy link

Here are the changes/additions that were made:

  1. PPO.py : Fixed Indentations and added support for Metal Performance Shaders
  2. .gitignore : For making sure unimportant files are not added
  3. PPO_preTrained/RocketLanding/PPO_RocketLanding_0_0.pth : Model checkpoint for fine tuning and better usability
  4. README.md :
    • updated video for better understanding of usage
    • Dependency guide
    • Apple Silicon support
  5. plot_graph.py : Minor fixes and better clarity
  6. requirements.txt : project lib requirements
  7. test.py :
    • Find and get the latest trained model checkpoint and run tests
    • Confetti celebration for close to perfect landing
  8. train.py :
    • Continue training from a checkpoint
    • Cleanups by refactoring/imports from utils.py
  9. utils.py :
    • Training Management: Set configurations as per user inputs
    • Checkpoint Management: Finding and loading checkpoints
    • Logging Management: Continue log updating if file exists
    • Plot Management: Plot setup and updating
    • fix: np.mat() --> np.asmatrix() latest version
    • Save Training and logs

…ore training;fix: add error handling for improved stability;feat: add user prompt for training options and enable Metal Performance Shaders;build: add requirements.txt for easier installation;feat: enhance test.py with addons and speed optimizations
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

1 participant