Home Explore Help
Register Sign In
david
/
flash-attention
mirror of https://github.com/Dao-AILab/flash-attention
1
0
Fork 0
Files Issues 0 Wiki
Tree: c272619bf5
Branches Tags
flash-attention
/
training
/
configs
/
callbacks
/
flop-count.yaml

flop-count.yaml 126 B
History Raw

12345 
  1. flop_count:
    2. 

  2.   _target_: src.callbacks.flop_count.FlopCount
    3. 

  3.   profilers: ['fvcore']
    4. 

  4.   input_size: [3, 224, 224]
    5. 

  5.   device: null
    6.