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

fix: 🐛 update requirements.txt #10

Open
wants to merge 2 commits into
base: master
Choose a base branch
from
Open
Show file tree
Hide file tree
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
37 changes: 37 additions & 0 deletions autofe_pai.yml
Original file line number Diff line number Diff line change
@@ -0,0 +1,37 @@
authorName: Xiang Long
experimentName: example-auto-auto-fe
# how many trials could be concurrently running
trialConcurrency: 1
# maximum experiment running duration
maxExecDuration: 3h
# empty means never stop
maxTrialNum: 100
# choice: local, remote, pai
trainingServicePlatform: pai
searchSpacePath: search_space.json
# choice: true, false
useAnnotation: false
tuner:
codeDir: .
classFileName: autofe_tuner.py
className: AutoFETuner
classArgs:
optimize_mode: maximize
trial:
command: python3 main.py
codeDir: .
gpuNum: 0
cpuNum: 1
memoryMB: 8196
image: msranni/nni:v1.9
virtualCluster: default
nniManagerNFSMountPath: <your_mount_dir>/nni_autofe_log
containerNFSMountPath: /mnt/<your_pai_storage_config_name>/nni_autofe_log
paiStorageConfigName: <your_pai_storage_config_name>
# Configuration to access OpenPAI Cluster
paiConfig:
userName: <your_pai_user_name>
token: <your_pai_application_token>
host: https://int.openpai.org
reuse: true # pai has job limit concurrency
reuse: true
4 changes: 2 additions & 2 deletions requirments.txt
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
lightgbm
pandas
pandas==0.25
numpy
sklearn
nni==0.9.1
nni==1.9
gensim