mirror of
https://github.com/microsoft/autogen.git
synced 2025-08-03 06:12:22 +00:00
19 lines
587 B
Python
19 lines
587 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] && ",
|
|
"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) |