Closed abiotik closed 3 weeks ago
I haven't looked very deeply into this, but it looks like you're using a very old version of tensorflow.
https://pypi.org/project/tensorflow-gpu/ mentions "tensorflow-gpu
has been removed. Please install tensorflow instead."
The latest published version of TF is 2.17.0: https://pypi.org/project/tensorflow/, and 2.18.0 should be available soon.
Consider Stack Overflow for getting support using TensorBoard—they have a larger community with better searchability:
Here is my script which I ran import torch import torch.nn as nn from torch.autograd import Variable import time as t import matplotlib.pyplot as plt import sys plt.switch_backend('agg') sys.path.append('/home/nikola/Desktop/tik/GENiPPI/gPPMol/models_3d') import os from torchvision import utils from tensorboard_logger import Logger from tqdm import tqdm import numpy as np SAVE_PER_TIMES = 1000 from torch.nn.utils.rnn import pack_padded_sequence,pad_packed_sequence
class Generator(torch.nn.Module): def init(self, channels,is_cuda): super().init()
Filters [1024, 512, 256]
class Discriminator(torch.nn.Module): def init(self, channels): super().init()
Filters [256, 512, 1024]
class WGAN_CP(object): def init(self, channels,is_cuda,generator_iters,gnn_interface_model,encoder,decoder,device,batch_size=64): print("WGAN_CP init model.") self.G = Generator(channels,is_cuda) self.D = Discriminator(channels) self.gnn_interface_model = gnn_interface_model self.encoder = encoder self.decoder = decoder self.device = device self.C = channels
and I got this error AttributeError: 'Logger' object has no attribute 'writer'