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

Add python train.py --freeze N argument #4238

Merged
merged 4 commits into from
Jul 30, 2021

Commits on Jul 30, 2021

  1. Add freeze as an argument

    I train on different platforms and sometimes I want to freeze some layers. I have to go into the code and change it and also keep track of how many layers I froze on what platform. Please add the number of layers to freeze as an argument in future versions thanks.
    IneovaAI committed Jul 30, 2021
    Configuration menu
    Copy the full SHA
    ab5c886 View commit details
    Browse the repository at this point in the history
  2. Update train.py

    IneovaAI committed Jul 30, 2021
    Configuration menu
    Copy the full SHA
    5dd8836 View commit details
    Browse the repository at this point in the history
  3. Update train.py

    IneovaAI committed Jul 30, 2021
    Configuration menu
    Copy the full SHA
    2b48f3d View commit details
    Browse the repository at this point in the history
  4. Cleanup

    glenn-jocher committed Jul 30, 2021
    Configuration menu
    Copy the full SHA
    e4a5a16 View commit details
    Browse the repository at this point in the history