wehub-resource-sync
25 lines
797 B
Plaintext
25 lines
797 B
Plaintext
name: mnist-autolog-example
|
|
|
|
python_env: python_env.yaml
|
|
|
|
entry_points:
|
|
main:
|
|
parameters:
|
|
max_epochs: {type: int, default: 5}
|
|
devices : {type: str, default: "auto"}
|
|
strategy: {type str, default: "auto"}
|
|
accelerator: {type str, default: "auto"}
|
|
batch_size: {type: int, default: 64}
|
|
num_workers: {type: int, default: 3}
|
|
learning_rate: {type: float, default: 0.001}
|
|
|
|
command: |
|
|
python mnist_autolog_example.py \
|
|
--trainer.max_epochs={max_epochs} \
|
|
--trainer.devices={devices} \
|
|
--trainer.strategy={strategy} \
|
|
--trainer.accelerator={accelerator} \
|
|
--data.batch_size={batch_size} \
|
|
--data.num_workers={num_workers} \
|
|
--model.learning_rate={learning_rate}
|