Skip to content

mewc-train v2.0.0
Compare
Choose a tag to compare
@zaandahl zaandahl released this 22 Aug 06:21
· 12 commits to main since this release
v2.0.0
f8e788e
This commit was created on GitHub.com and signed with GitHub’s verified signature.
GPG key ID: B5690EEEBB952194
Learn about vigilant mode.

Version 2 Updates

The mewc-train Docker image has been updated to version 2. Key updates include:

  • Base Image: Uses the new mewc-flow base image featuring tensorflow/tensorflow:2.16.1-gpu, CUDA, cuDNN, and JAX.
  • Hugging Face Base Models: Start training off a base model from Hugging Face.
  • Improved Training Control: New user configurable options to fine tune model training.
  • Optional Validation Path: There is an option to specify separate paths for data|validation VAL_PATH and data|test TEST_PATH. Doing so will keep test data isolated from hyper-parameter tuning leakage.
Assets 2
Loading