search

teamtomo / membrain-seg

membrane segmentation in 3D for cryo-ET
Other
48 stars 12 forks source link

improve logging on Unet #2

Closed kevinyamauchi closed 1 year ago

kevinyamauchi commented 1 year ago

this PR adds some improved logging to the UNet by writing to the LightningModule logger the summary statistics of the validation step (both metric and loss).

alisterburt commented 1 year ago

not 100% how this improves logging but happy for you to merge if you're happy!

kevinyamauchi commented 1 year ago

@alisterburt, sorry for the vague description. Basically, I just added a step that logs the summary statistics of the validation loss and metric (i.e., mean over all batches). When we attach a tensorboard logger, these will be viewable as realtime plots there.