llbbcc

Results 3 issues of llbbcc

Hi! I have finished the training process on MNIST and FashionMNIST datasets, but when I set the dataset parameters to celeba-gender-train, I get the following error No such file or...

We can run the code on the GPU with the following modifications: ```bash pip install onnxruntime-gpu ``` Modify the following code segment in `./imwatermark/rivaGAN.py` ``` RivaWatermark.encoder = onnxruntime.InferenceSession( os.path.join(modelDir, 'rivagan_encoder.onnx'))...

Thanks for your codes! When I train the model with multiple GPUs with `net_model = torch.nn.DataParallel(net_model)`, images obtained from sampling are noise images. I load the model with `model =...