PrakhAI commited on
Commit
11f19f6
·
1 Parent(s): 1adf129

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -9,7 +9,9 @@ with fs.open("PrakhAI/HelloWorld/checkpoint.msgpack", "rb") as f:
9
 
10
  print(type(state))
11
  print(state)
 
12
  print(state['params'])
 
13
 
14
  # logits = state.apply_fn({'params': state.params}, batch['image'])
15
  # print(logits)
 
9
 
10
  print(type(state))
11
  print(state)
12
+ print(state.keys)
13
  print(state['params'])
14
+ print(state['opt_state'])
15
 
16
  # logits = state.apply_fn({'params': state.params}, batch['image'])
17
  # print(logits)