Hennara commited on
Commit
166583b
1 Parent(s): 38f4f24

add utils file

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -37,6 +37,7 @@ def main():
37
  intermediate_size)
38
 
39
  dense_model_memory = layers * (attention_memory + dense_mlp_memory) // (1024 ** 3)
 
40
  space = st.empty()
41
  space.markdown('<div style="height: 20px;"></div>', unsafe_allow_html=True)
42
 
 
37
  intermediate_size)
38
 
39
  dense_model_memory = layers * (attention_memory + dense_mlp_memory) // (1024 ** 3)
40
+ st.write("This estimation is for the training phase, all computation done for AdamW optimizer.")
41
  space = st.empty()
42
  space.markdown('<div style="height: 20px;"></div>', unsafe_allow_html=True)
43