The model is available on GitHub.It is for manipulation of multiple face attributes.
Here is the link

  
      
      GitHub
  
  
    

Prinsphield/ELEGANT 3

ELEGANT: Exchanging Latent Encodings with GAN for Transferring Multiple Face Attributes - Prinsphield/ELEGANT


  
  
    
    
  
  


It generates four files
Enc-iter.pth
Dec-iter.pth
        D1-iter.pth

        D2-iter.pth

torch_out=torch.onnx.export(model,x,model.onnx, verbose=True)
Export function gives the graph but when I print torch_out it gives None but it should print tensor value
And this is causing AtrributError when comparing torch_out with caffe2 model
np.testing.assert_almost_equal(torch_out.data.cpu().numpy(), c2_out, decimal=3)
In this I am getting AtrributError
Nonetype object has no attribute ‘data’
I have asked it on GitHub/onnx but they are saying ask this on pytorch forum .issue is in pytorch code.