MIPT-Oulu / Collagen

Deep Learning Framework for Reproducible Experiments
MIT License
1 stars 0 forks source link

Batch-wise logging to tensorboard #33

Open lext opened 5 years ago

lext commented 5 years ago

Important for GAN training. If it is possible to do this already, add to #28

hoanghng commented 5 years ago

Do you mean this https://github.com/MIPT-Oulu/Collagen/blob/4fdf72be9d90e95907cebcf0196fb4eda6996215/collagen/logging/_logging.py#L16

It's a strategy-level callback that logs all metrics to tensorboard.

hoanghng commented 5 years ago

For visualization, use this https://github.com/MIPT-Oulu/Collagen/blob/master/collagen/callbacks/_visualizers.py

lext commented 5 years ago

A modification of https://github.com/MIPT-Oulu/Collagen/blob/1391ce9edcd2ed84918d0700758421f716551548/collagen/callbacks/_visualizers.py#L82 should do

lext commented 5 years ago

@hoanghng What is the status of this?