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 NN Template #44

Draft
wants to merge 4 commits into
base: main
Choose a base branch
from
Draft
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
59 changes: 59 additions & 0 deletions configs/GrokAi/nn-template.yaml
Original file line number Diff line number Diff line change
@@ -0,0 +1,59 @@
target_repository:
HTTPS: https://github.com/grok-ai/nn-template
# OPTIONAL, for some protected repository
# username: my-nick
# OPTIONAL, paired with the username
# password: dont-tell-anyone
# OPTIONAL, overrides the user/pass
# token: authentication-token
# OPTIONAL, checkout a particular branch or a tag
checkout: master
# OPTIONAL, define installing package extras
install_extras: all
# OPTIONAL, install additional requirements from a file
requirements_file: requirements.txt
# copy some test from the target repository
copy_tests:
- integrations
- tests/__init__.py
- tests/helpers

contact:
slack:
- lucmos
email: [[email protected]]

# OPTIONAL, running before installing your project
# WARNING: this is custom for given OS, we do not manage any inter OS compatibility
before_install:
- pip install Cython==3.56

# OPTIONAL, if any installation require some env. variables
env:
HOROVOD_BUILD_ARCH_FLAGS: "-mfma"
# HOROVOD_WITHOUT_MXNET: 1
# HOROVOD_WITHOUT_TENSORFLOW: 1

dependencies:
- name: pytorch-lightning
HTTPS: https://github.com/PyTorchLightning/pytorch-lightning.git
checkout: release/1.5.x
# install_extras: all
- name: Cython
checkout: 0.29.24

# OPTIONAL, running before installing your project
# WARNING: this is custom for given OS, we do not manage any inter OS compatibility
before_test:
- wget https://pl-flash-data.s3.amazonaws.com/titanic.zip

testing:
dirs:
- tests
# OPTIONAL, additional pytest arguments
pytest_args: --strict

runtimes:
- {os: "ubuntu-20.04", python-version: "3.9"}
- {os: "macOS-10.15", python-version: "3.7"}
- {os: "windows-2019", python-version: "3.8"}