mirror of
https://github.com/microsoft/FLAML.git
synced 2026-02-09 10:19:18 +08:00
3de0dc667e
* lint * bump ver * bump ver * fixed circular import --------- Co-authored-by: Jirka Borovec <6035284+Borda@users.noreply.github.com>
22 lines
595 B
Python
22 lines
595 B
Python
from azureml.core import Experiment, ScriptRunConfig, Workspace
|
|
|
|
ws = Workspace.from_config()
|
|
|
|
compute_target = ws.compute_targets["V100-4"]
|
|
# compute_target = ws.compute_targets['K80']
|
|
command = [
|
|
"pip install torch transformers datasets flaml[blendsearch,ray] && ",
|
|
"python test_electra.py",
|
|
]
|
|
|
|
config = ScriptRunConfig(
|
|
source_directory="hf/",
|
|
command=command,
|
|
compute_target=compute_target,
|
|
)
|
|
|
|
exp = Experiment(ws, "test-electra")
|
|
run = exp.submit(config)
|
|
print(run.get_portal_url()) # link to ml.azure.com
|
|
run.wait_for_completion(show_output=True)
|