adamcasson commited on
Commit
6ecfb5c
·
1 Parent(s): 64e5a86

add markdown title

Browse files
Files changed (1) hide show
  1. app.py +4 -0
app.py CHANGED
@@ -42,6 +42,10 @@ def calculator(n_layer, d_model, n_heads, n_vocab, n_ctx, ff_ratio, incl_embed):
42
 
43
 
44
  with gr.Blocks() as iface:
 
 
 
 
45
  with gr.Row():
46
  with gr.Column():
47
  n_layer = gr.Number(label="Number of layers (n_layer)")
 
42
 
43
 
44
  with gr.Blocks() as iface:
45
+ gr.Markdown("## Transformer FLOPs Calculator")
46
+ gr.Markdown(
47
+ "Calculate how many FLOPs a Transformer language model has using the method described in [DeepMind's Chinchilla scaling law paper](https://arxiv.org/abs/2203.15556) (see Appendix F)."
48
+ )
49
  with gr.Row():
50
  with gr.Column():
51
  n_layer = gr.Number(label="Number of layers (n_layer)")