Giters

pyronear / pyro-vision

Computer vision library for wildfire detection 🌲 Deep learning models in PyTorch & ONNX for inference on edge devices (e.g. Raspberry Pi)

Home Page:https://pyronear.org/pyro-vision/

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

[references] Disable codecarbon logger

frgfm opened this issue · comments

commented

Running the classification training script with any options gets the codecarbon logger to start. But this becomes a bit too much as it pops information in the middle of training cycles:

Namespace(amp=False, arch='rexnet1_0x', batch_size=32, data_path='/home/fg/Downloads/', dataset='openfire', device=None, epochs=20, find_lr=False, freeze_until=None, img_size=224, lr=0.001, name=None, norm_wd=None, opt='adamp', output_file='./model.pth', pretrained=False, resume='', sched='onecycle', show_samples=True, test_only=False, wb=False, weight_decay=0, workers=16)
Loading data
  3%|██▏                                                                           | 225/7880 [00:13<05:50, 21.84it/s][codecarbon INFO @ 23:26:55] Energy consumed for RAM : 0.000024 kWh. RAM Power : 5.744381904602051 W
[codecarbon INFO @ 23:26:55] Energy consumed for all CPUs : 0.000000 kWh. All CPUs Power : 0.0 W
[codecarbon INFO @ 23:26:55] 0.000024 kWh of electricity used since the begining.
  6%|████▌                                                                         | 457/7880 [00:28<07:48, 15.84it/s][codecarbon INFO @ 23:27:10] Energy consumed for RAM : 0.000048 kWh. RAM Power : 5.744381904602051 W
[codecarbon INFO @ 23:27:10] Energy consumed for all CPUs : 0.000000 kWh. All CPUs Power : 0.0 W
[codecarbon INFO @ 23:27:10] 0.000048 kWh of electricity used since the begining.
  8%|█████▉                                                                        | 598/7880 [00:42<08:30, 14.26it/s][codecarbon INFO @ 23:27:25] Energy consumed for RAM : 0.000072 kWh. RAM Power : 5.744381904602051 W
[codecarbon INFO @ 23:27:25] Energy consumed for all CPUs : 0.000000 kWh. All CPUs Power : 0.0 W
[codecarbon INFO @ 23:27:25] 0.000072 kWh of electricity used since the begining.
 10%|███████▌                                                                      | 762/7880 [00:57<09:24, 12.60it/s][codecarbon INFO @ 23:27:40] Energy consumed for RAM : 0.000096 kWh. RAM Power : 5.744381904602051 W
[codecarbon INFO @ 23:27:40] Energy consumed for all CPUs : 0.000000 kWh. All CPUs Power : 0.0 W
[codecarbon INFO @ 23:27:40] 0.000096 kWh of electricity used since the begining.
 11%|████████▉                                                                     | 899/7880 [01:13<08:53, 13.09it/s][codecarbon INFO @ 23:27:55] Energy consumed for RAM : 0.000120 kWh. RAM Power : 5.744381904602051 W
[codecarbon INFO @ 23:27:55] Energy consumed for all CPUs : 0.000000 kWh. All CPUs Power : 0.0 W
[codecarbon INFO @ 23:27:55] 0.000120 kWh of electricity used since the begining.
 14%|██████████▍

I think it's better to either set a logger level that will prevent this or disable the logger completely