NVlabs / few-shot-vid2vid

Pytorch implementation for few-shot photorealistic video-to-video translation.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Have an error when using tensorboard

HappyDeepLearning opened this issue · comments

(im_py36) mjz@8GPUServer:~/Python_workspace/few-shot-vid2vid$ bash scripts/pose/train_g2.sh
CustomDatasetDataLoader
606 sequences
dataset [PoseDataset] was created
create web directory ./checkpoints/136video/pose/web...
Warning: using Python fallback for SyncBatchNorm, possibly because apex was installed without --cuda_ext. The exception raised when attempting to import the cuda backend was: No module named 'syncbn'
---------- Networks initialized -------------
---------- Optimizers initialized -------------
model [Vid2VidModel] was created
/home/mjz/anaconda3/envs/im_py36/lib/python3.6/site-packages/torch/nn/functional.py:2479: UserWarning: Default upsampling behavior when mode=bilinear is changed to align_corners=False since 0.4.0. Please specify align_corners=True if the old behavior is desired. See the documentation of nn.Upsample for details.
"See the documentation of nn.Upsample for details.".format(mode))
(epoch: 1, iters: 600, time: 0.283) G_GAN: 1.044 G_GAN_Feat: 1.289 G_VGG: 6.623 Gf_GAN: 8.425 Gf_GAN_feat: 39.303 F_Flow: 12.360 F_Warp: 4.596 F_Mask: 6.243 D_real: 0.188 D_fake: 1.958 Df_real: 9.018 Df_fake: 10.353
Traceback (most recent call last):
File "/home/mjz/anaconda3/envs/im_py36/lib/python3.6/site-packages/PIL/ImageFile.py", line 498, in _save
fh = fp.fileno()
io.UnsupportedOperation: fileno

During handling of the above exception, another exception occurred:

Traceback (most recent call last):
File "train.py", line 77, in
train()
File "train.py", line 66, in train
if trainer.end_of_iter(loss_dict, generated + data_list + data_ref_list, model):
File "/home/mjz/Python_workspace/few-shot-vid2vid/models/trainer.py", line 67, in end_of_iter
self.visualizer.display_current_results(visuals, epoch, total_steps)
File "/home/mjz/Python_workspace/few-shot-vid2vid/util/visualizer.py", line 105, in display_current_results
scipy.misc.toimage(image_numpy).save(s, format="jpeg")
File "/home/mjz/anaconda3/envs/im_py36/lib/python3.6/site-packages/PIL/Image.py", line 2084, in save
save_handler(self, fp, filename)
File "/home/mjz/anaconda3/envs/im_py36/lib/python3.6/site-packages/PIL/JpegImagePlugin.py", line 770, in _save
ImageFile._save(im, fp, [("jpeg", (0, 0) + im.size, 0, rawmode)], bufsize)
File "/home/mjz/anaconda3/envs/im_py36/lib/python3.6/site-packages/PIL/ImageFile.py", line 513, in _save
fp.write(d)
TypeError: string argument expected, got 'bytes'