search

valohai / valohai-utils

Python helper library for Valohai
MIT License
2 stars 2 forks source link

Can't define the environment in valohai.prepare() #92

Closed drazendee closed 2 years ago

drazendee commented 2 years ago

It would be great if I could define the default environment for the step inside valohai.prepare()

Something like valohai.prepare(step="train", image="tensorflow/tensorflow:2.6.0", environment="myorg-aws-eu-west-1-p2.xlarge")