Skip to content

Commit f09d5f4

Browse files
author
Anagha Jamthe
committed
fine tuning Tapis stuff added
1 parent 025b08a commit f09d5f4

1 file changed

Lines changed: 7 additions & 7 deletions

File tree

tutorials/Fine_Tuning_Vision/01-intro-fine-tuning.md

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
## Fine Tuning Vision Models Using Ultralytics and Tapis
1+
## Fine-Tuning Vision Models Using Ultralytics and Tapis
22

33
[Lecture Slides](https://docs.google.com/presentation/d/1gykJ0awIswnAUU-qob3MnzGD_5a729RJVnMhknykeVs/edit?usp=sharing)
44

@@ -19,7 +19,7 @@ The following JSON represents the application definition that was used to regist
1919
{
2020
"id": "ultralytics-fine-tune",
2121
"version": "0.1",
22-
"description": "An app to fine-tune ultralytics Yolo using Singularity in batch mode.",
22+
"description": "An app to fine-tune ultralytics YOLO using Singularity in batch mode.",
2323
"jobType": "BATCH",
2424
"runtime": "SINGULARITY",
2525
"containerImage": "/work/projects/aci/cic/apps/ultralytics-fine-tune/Ultralytics_FT_Tapis_app.sif",
@@ -49,7 +49,7 @@ The following JSON represents the application definition that was used to regist
4949
"memoryMB": 1,
5050
"nodeCount": 1,
5151
"coresPerNode": 1,
52-
"maxMinutes": 10
52+
"maxMinutes": 60
5353
}
5454
}
5555
</code></pre>
@@ -102,7 +102,7 @@ Navigate to the **Apps** list and select the `ultralytics-fine-tune` app. Click
102102
![Step 1 - Selecting the Ultralytics App in UI](/tutorials/images/Step1-Ultralyticsapp.png)
103103

104104
### Step 2: Edit the JSON Payload
105-
Paste the job JSON provided above into the editor and click `Submit`.
105+
Paste the job JSON provided below into the editor and click `Submit`.
106106

107107
<div style="max-height:400px; overflow:auto; border:1px solid #ddd; padding:10px;">
108108

@@ -155,19 +155,19 @@ These are specific flags passed to the Slurm scheduler on the Vista system:
155155
After clicking **Submit**, navigate to the **Jobs** tab. You can monitor the status as it moves from `PENDING` to `RUNNING` and finally `FINISHED`.
156156

157157
![Step 3 - Monitoring Job Status in the Dashboard](/tutorials/images/fine-tune-job-running.png)
158-
Since we are running 100 Epochs to fine-tune, this will take around 10-12 mins to finish.
158+
Since we are running 100 Epochs to fine-tune, this will take around 10-12 minutes to finish.
159159

160160

161161
### Step 4: Job Output and Results location
162162
Once the job finishes, you should see output similar to the image below.
163163
![Step 4 - After Job completes](/tutorials/images/fine-tune-job-completion.png)
164164

165-
You should see a train directory with weights. Inside weights, you can access the best.pt model. This file will be accessible to you Jupyter.
165+
You should see a train directory with weights. Inside the `weights` directory, you can access the `best.pt model`. This file will be accessible in Jupyter Notebook.
166166

167167

168168
### Step 5: Finding best.pt file from Jupyter
169169

170-
Inside your work directory, you should see a vista folder. In there you can find the directory named with your job uuid. In the above image the job uuid is highlighted. In the job directory you will find the outputs of the job archieved with the same train directory containing the best.pt file.
170+
Inside your work directory, you should see a `vista` folder. In there you can find the directory named with your `jobUUID`. In the above image the job uuid is highlighted. In the job directory you will find the outputs of the job archived with the same train directory containing the best.pt file.
171171
`$WORK -> vista -> jobUUID -> train -> weights -> best.pt`
172172

173173

0 commit comments

Comments
 (0)