flop_count:
  _target_: src.callbacks.flop_count.FlopCount
  profilers: ['fvcore']
  input_size: [3, 224, 224]
  device: null
