Merge pull request #87 from Azure/patch-1-patch-brackets

Fix brackets in the train.yaml file
This commit is contained in:
Scott Donohoo 2022-09-21 07:56:43 -05:00 коммит произвёл GitHub
Родитель db806dc661 472bd80d65
Коммит 7a7324416a
Не найден ключ, соответствующий данной подписи
Идентификатор ключа GPG: 4AEE18F83AFDEB23
1 изменённых файлов: 13 добавлений и 13 удалений

Просмотреть файл

@ -101,22 +101,22 @@ command: >-
python train.py
--train_images ${{inputs.train_images}}
--valid_images ${{inputs.valid_images}}
[--batch_size ${{inputs.batch_size}}]
[--num_workers ${{inputs.num_workers}}]
[--prefetch_factor ${{inputs.prefetch_factor}}]
[--persistent_workers ${{inputs.persistent_workers}}]
[--pin_memory ${{inputs.pin_memory}}]
[--non_blocking ${{inputs.non_blocking}}]
[--model_arch ${{inputs.model_arch}}]
[--model_arch_pretrained ${{inputs.model_arch_pretrained}}]
[--num_epochs ${{inputs.num_epochs}}]
[--learning_rate ${{inputs.learning_rate}}]
[--momentum ${{inputs.momentum}}]
$[[--batch_size ${{inputs.batch_size}}]]
$[[--num_workers ${{inputs.num_workers}}]]
$[[--prefetch_factor ${{inputs.prefetch_factor}}]]
$[[--persistent_workers ${{inputs.persistent_workers}}]]
$[[--pin_memory ${{inputs.pin_memory}}]]
$[[--non_blocking ${{inputs.non_blocking}}]]
$[[--model_arch ${{inputs.model_arch}}]]
$[[--model_arch_pretrained ${{inputs.model_arch_pretrained}}]]
$[[--num_epochs ${{inputs.num_epochs}}]]
$[[--learning_rate ${{inputs.learning_rate}}]]
$[[--momentum ${{inputs.momentum}}]]
--model_output ${{outputs.trained_model}}
--checkpoints ${{outputs.checkpoints}}
[--register_model_as ${{inputs.register_model_as}}]
$[[--register_model_as ${{inputs.register_model_as}}]]
--enable_profiling ${{inputs.enable_profiling}}
[--multiprocessing_sharing_strategy ${{inputs.multiprocessing_sharing_strategy}}]
$[[--multiprocessing_sharing_strategy ${{inputs.multiprocessing_sharing_strategy}}]]
distribution:
# NOTE: using type:pytorch will use all the right env variables for pytorch init_process_group
type: pytorch