Update app.py
Browse files
app.py
CHANGED
|
@@ -49,12 +49,36 @@ def compile_model(model_name, vmem_value, lpmem_value):
|
|
| 49 |
print(results)
|
| 50 |
|
| 51 |
output_text = ''
|
| 52 |
-
|
| 53 |
-
|
|
|
|
|
|
|
|
|
|
| 54 |
else:
|
| 55 |
-
|
| 56 |
-
|
| 57 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 58 |
|
| 59 |
return output_text
|
| 60 |
|
|
|
|
| 49 |
print(results)
|
| 50 |
|
| 51 |
output_text = ''
|
| 52 |
+
|
| 53 |
+
if results['cycles_npu'] == 0:
|
| 54 |
+
output_text = 'FAILURE cannot optimize TFLITE model for SR100'
|
| 55 |
+
output_text += '<br><br>Compiler log<br>---------------------------------------'
|
| 56 |
+
output_text += f'<br>{results["vela_log"]'
|
| 57 |
else:
|
| 58 |
+
if success:
|
| 59 |
+
output_text = 'SUCCESS, model fits on SR100'
|
| 60 |
+
else:
|
| 61 |
+
output_text = 'FAILURE, model does not fit on SR100'
|
| 62 |
+
|
| 63 |
+
weights_size = results['weights_size'] / 1024.0
|
| 64 |
+
arena_size = results['arena_cache_size'] / 1024.0
|
| 65 |
+
clock = results['core_clock'] / 1.0e6
|
| 66 |
+
|
| 67 |
+
output_text += f'<br>clock = {clock:0.1f} MHz'
|
| 68 |
+
output_text += f'<br>Model size = {weights_size} KiB'
|
| 69 |
+
output_text += f'<br>Arena cache size = {arena_size} KiB'
|
| 70 |
+
output_text += f'<br>model loc {results["model_loc"]'
|
| 71 |
+
#output_text += f'<br>System config {results["system_config"]'
|
| 72 |
+
|
| 73 |
+
|
| 74 |
+
vmem_size = results['vmem_size'] / 1024.0
|
| 75 |
+
lpmem_size = results['lpmem_size'] / 1024.0
|
| 76 |
+
vmem_perc = results['vmem_size'] * 100.0 / results['vmem_size_limit']
|
| 77 |
+
lpmem_perc = results['lpmem_size'] * 100.0 / results['lpmem_size_limit']
|
| 78 |
+
|
| 79 |
+
output_text += f'<br>vmem_size = {vmem_size} KiB ({vmem_perc:0.1f}%)'
|
| 80 |
+
output_text += f'<br>lpmem_size = {lpmem_size} KiB ({lpmem_perc:0.1f}%)'
|
| 81 |
+
|
| 82 |
|
| 83 |
return output_text
|
| 84 |
|