Spaces:
Runtime error
Runtime error
Commit
·
88aa939
1
Parent(s):
2133880
sharing last
Browse files
app.py
CHANGED
|
@@ -29,9 +29,9 @@ with st.expander("More options"):
|
|
| 29 |
batch_size = int(st.number_input('Microbatch size (sequences)', min_value=1, step=1, value=1, format="%i"))
|
| 30 |
precisions_names = ('Full', 'Mixed ("O1")', 'Pure 16-bit')
|
| 31 |
precisions_values = ('O0', 'O1', 'O3')
|
|
|
|
| 32 |
sharing_groups = int(st.number_input('Shared parameter groups (used if Share parameters is checked)',
|
| 33 |
min_value=1, step=1, value=1, format="%i"))
|
| 34 |
-
precision = st.selectbox('Precision', precisions_names, index=1)
|
| 35 |
|
| 36 |
args = mem_calc.parse_args(f"""
|
| 37 |
--model {model} --optimizer {optimizers_values[optimizers_names.index(optimizer)]}
|
|
|
|
| 29 |
batch_size = int(st.number_input('Microbatch size (sequences)', min_value=1, step=1, value=1, format="%i"))
|
| 30 |
precisions_names = ('Full', 'Mixed ("O1")', 'Pure 16-bit')
|
| 31 |
precisions_values = ('O0', 'O1', 'O3')
|
| 32 |
+
precision = st.selectbox('Precision', precisions_names, index=1)
|
| 33 |
sharing_groups = int(st.number_input('Shared parameter groups (used if Share parameters is checked)',
|
| 34 |
min_value=1, step=1, value=1, format="%i"))
|
|
|
|
| 35 |
|
| 36 |
args = mem_calc.parse_args(f"""
|
| 37 |
--model {model} --optimizer {optimizers_values[optimizers_names.index(optimizer)]}
|