pai/examples/mxnet/mxnet.autoencoder.json

15 строки
352 B
JSON

{
"jobName": "mxnet-autoencoder",
"image": "openpai/pai.example.mxnet:stable",
"taskRoles": [
{
"name": "main",
"taskNumber": 1,
"cpuNumber": 4,
"memoryMB": 8192,
"gpuNumber": 1,
"command": "pip install scipy scikit-learn && cd incubator-mxnet/example/autoencoder && python mnist_sae.py --gpu"
}
]
}