Skip to content

Commit

Permalink
qGAN - PyTorch Discriminator Fix (qiskit-community/qiskit-aqua#1391)
Browse files Browse the repository at this point in the history
* update pytorch discriminator

Co-authored-by: Steve Wood <40241007+woodsp-ibm@users.noreply.github.com>
Co-authored-by: woodsp <woodsp@us.ibm.com>
  • Loading branch information
3 people authored Oct 26, 2020
1 parent 326684d commit 799b2f8
Showing 1 changed file with 1 addition and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -90,7 +90,7 @@ def load_model(self, load_dir: str):
Args:
load_dir: file with stored pytorch discriminator model to be loaded
"""
torch.load(load_dir)
self._discriminator = torch.load(load_dir)

@property
def discriminator_net(self):
Expand Down

0 comments on commit 799b2f8

Please sign in to comment.