search

Shreeyak / pytorch-lightning-segmentation-template

Semantic Segmentation on the LaPa dataset using Pytorch Lightning
Apache License 2.0
13 stars 5 forks source link

add python logging and create custom rank_zero_info/debug/warn #56

Open Shreeyak opened 3 years ago

Shreeyak commented 3 years ago

Using pytorch lightning's python logging func has weird print statements. Use custom python logging and wrap it with @rank_zero_only decorator.

/home/shrek/anaconda3/envs/cleargrasp/lib/python3.8/site-packages/pytorch_lightning/utilities/distributed.py:49: UserWarning: WARN: LogMedia only works with wandb logger. Current logger: None. Will not log any media to wandb this run
  warnings.warn(*args, **kwargs)