diff options
Diffstat (limited to 'simulation/vi_blocks.py')
| -rw-r--r-- | simulation/vi_blocks.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/simulation/vi_blocks.py b/simulation/vi_blocks.py index 94038a5..1177979 100644 --- a/simulation/vi_blocks.py +++ b/simulation/vi_blocks.py @@ -51,8 +51,8 @@ def create_vi_model(n_nodes, n_samp=100): if __name__ == "__main__": - n_cascades = 10000 - batch_size = 1000 + #n_cascades = 10000 + batch_size = 10 n_samples = 50 graph = utils.create_random_graph(n_nodes=4) print('GRAPH:\n', graph, '\n-------------\n') |