Skip to content

Commit 24b177e

Browse files
committed
Convert random seed to int data type.
1 parent 2bf65ba commit 24b177e

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

examples/sparse_regression.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -295,7 +295,7 @@ def main(args):
295295
with torch.no_grad():
296296
init_losses.append(loss_fn(model, guide, X, Y, hypers).item())
297297

298-
pyro.set_rng_seed(np.argmin(init_losses))
298+
pyro.set_rng_seed(int(np.argmin(init_losses)))
299299
pyro.clear_param_store()
300300
guide = AutoDelta(model, init_loc_fn=init_loc_fn)
301301

0 commit comments

Comments
 (0)