Spaces:

porestar
/

PorousMediaGAN

Runtime error

lmoss commited on Jan 17, 2022

Commit

2444ae1

1 Parent(s): a577b73

init

Files changed (1) hide show

app.py CHANGED Viewed

@@ -18,7 +18,7 @@ pl = pv.Plotter(shape=(1, 1),
                      window_size=(800, 800))
 netG = DCGAN3D_G(64, 512, 1, 32, 1)
-netG.load_state_dict(torch.load("./src/berea_generator_epoch_24.pth"))
 z = torch.randn(1, 512, 5, 5, 5)
 with torch.no_grad():
     X = netG(z)

                      window_size=(800, 800))
 netG = DCGAN3D_G(64, 512, 1, 32, 1)
+netG.load_state_dict(torch.load("berea_generator_epoch_24.pth"))
 z = torch.randn(1, 512, 5, 5, 5)
 with torch.no_grad():
     X = netG(z)