Spaces:
Runtime error
Runtime error
Commit
•
ca1f064
1
Parent(s):
58ad9c8
Clearer message
Browse files
app.py
CHANGED
@@ -68,7 +68,7 @@ def count_files(*inputs):
|
|
68 |
Training_Steps = file_counter*200*2
|
69 |
else:
|
70 |
Training_Steps = file_counter*200
|
71 |
-
return([gr.update(visible=True), gr.update(visible=True, value=f"You are going to train {concept_counter} {type_of_thing}(s), with {file_counter} images for {Training_Steps} steps.
|
72 |
|
73 |
def train(*inputs):
|
74 |
torch.cuda.empty_cache()
|
|
|
68 |
Training_Steps = file_counter*200*2
|
69 |
else:
|
70 |
Training_Steps = file_counter*200
|
71 |
+
return([gr.update(visible=True), gr.update(visible=True, value=f"You are going to train {concept_counter} {type_of_thing}(s), with {file_counter} images for {Training_Steps} steps. The training should take around {round(Training_Steps/1.1, 2)} seconds, or {round((Training_Steps/1.1)/60, 2)} minutes. The setup, compression and uploading the model can take up to 20 minutes. As the T4-Small GPU costs US$0.60 for 1h, the estimated cost for this training is below US${round((((Training_Steps/1.1)/3600)+0.3+0.1)*0.60, 2)}. If you check the box below the GPU attribution will automatically removed after training is done and the model is uploaded. If not don't forget to come back here and swap the hardware back to CPU.")])
|
72 |
|
73 |
def train(*inputs):
|
74 |
torch.cuda.empty_cache()
|