pix2pixHD icon indicating copy to clipboard operation
pix2pixHD copied to clipboard

Published 20 hours ago • verified publisher icon NVIDIA

To make the tensorboard writer compatible to tf2

Open seriousran opened this issue 3 years ago • 2 comments

To make the tensorboard writer compatible to tf2 in visualizer.py

  1. version check for tf2 or not
  2. FileWriter to create_file_writer and following changes

seriousran avatar Nov 24 '21 03:11 seriousran

Hi! Thanks for updating this. But I encountered a problem that whenever tensorboard logging is enabled with this pull, an exception is thrown "Expected all tensors to be on the same device, but found at least two devices: cuda:0 and cuda:1".

I was training only with cuda:1.

Zephyr69 avatar Oct 23 '22 05:10 Zephyr69

Hi! Thanks for updating this. But I encountered a problem that whenever tensorboard logging is enabled with this pull, an exception is thrown "Expected all tensors to be on the same device, but found at least two devices: cuda:0 and cuda:1".

I was training only with cuda:1.

Nvm. Solved it.

Zephyr69 avatar Oct 26 '22 12:10 Zephyr69