mirror of
https://github.com/microsoft/autogen.git
synced 2025-08-02 05:42:13 +00:00
19 lines
610 B
Python
19 lines
610 B
Python
|
from azureml.core import Workspace, Experiment, ScriptRunConfig
|
||
|
|
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] ax-platform sqlalchemy && ",
|
||
|
|
"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)
|