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
|