Spaces:

apsys
/

normflows

Runtime error

apsys commited on Mar 18, 2023

Commit

0595c23

•

1 Parent(s): d409f21

push

Files changed (1) hide show

app.py CHANGED Viewed

@@ -22,8 +22,7 @@ def compute(dim):
     for idx in api.train(iters=iters):
         my_bar.progress(idx[0]/iters)
     my_bar.progress(100)
-    samples = np.array(api.model.sample(
-            torch.tensor(api.scaled).float()).detach())

     for idx in api.train(iters=iters):
         my_bar.progress(idx[0]/iters)
     my_bar.progress(100)
+    samples = np.delete(np.array(api.model.sample(torch.tensor(api.scaled).float()).detach()),np.argmin(np.array(api.model.sample(torch.tensor(api.scaled).float()).detach()),axis=0),0)