6 lines
126 B
YAML
6 lines
126 B
YAML
flop_count:
|
|
_target_: src.callbacks.flop_count.FlopCount
|
|
profilers: ['fvcore']
|
|
input_size: [3, 224, 224]
|
|
device: null
|