| | env_defaults: |
| |
|
| | SHARED_CMD_ARGS: ' |
| | -m src.train |
| | train_data=[vg-densecap-region_descriptions] eval_data=[vg-densecap-region_descriptions] |
| | +model=base_sam_captioner |
| | training.do_train=False |
| | training.do_eval=False |
| | training.do_inference=True |
| | training.num_masks_per_sample=1 |
| | +data.streaming=False |
| | training.max_eval_samples=10 |
| | training.max_train_samples=1 |
| | training.num_train_epochs=10 |
| | training.fp16=True |
| | training.output_dir=$AMLT_OUTPUT_DIR |
| | training.output_log_dir=$AMLT_LOGS_DIR |
| | model.cache_dir=/mnt/blob/weights/.model.cache/ |
| | training.dataloader_num_workers=4 |
| | ' |
| |
|
| |
|
| |
|
| | environment: |
| | image: nvidia/pytorch:23.07-py3 |
| | registry: nvcr.io |
| |
|
| | code: |
| | local_dir: $CONFIG_DIR/../ |
| |
|
| |
|
| |
|
| | jobs: |
| | - name: sam_captioner-infer-debug |
| | sku: G$NUM_GPUS |
| | preemptible: False |
| | process_count_per_node: 1 |
| | command: |
| | - . amlt_configs/setup.sh |
| | - source ~/.bashrc |
| | - . amlt_configs/setup_accelerate_on_azure.sh |
| | - . amlt_configs/post_process.sh |
| | |
| |
|
| | |
| | submit_args: |
| | env: |
| | AZFUSE_USE_FUSE: "1" |
| | SHARED_MEMORY_PERCENT: 0.5 |
| | container_args: |
| | shm_size: 256g |
