Agents limited to <10m

[cpranav93]

Currently the model cannot run for 10m agents due to the following error:

performing step 1 of 100
Traceback (most recent call last):
File "/gpfs/work1/0/einf4446/Pranav/DGL-PTM/dgl_ptm/dgl_ptm/model/initialize_model.py", line 327, in step
ptm_step(self.model_graph,self.model_data,self.step_count,self.steering_parameters)
File "/gpfs/work1/0/einf4446/Pranav/DGL-PTM/dgl_ptm/dgl_ptm/model/step.py", line 33, in ptm_step
local_attachment(agent_graph, n_FoF_links = 1, edge_prop = 'weight', p_attach=1. )
File "/gpfs/work1/0/einf4446/Pranav/DGL-PTM/dgl_ptm/dgl_ptm/network/local_attachment.py", line 12, in local_attachment
src_ids, dst_ids, _ = select_edges(graph,n_FoF_links,edge_prop)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/gpfs/work1/0/einf4446/Pranav/DGL-PTM/dgl_ptm/dgl_ptm/network/local_attachment.py", line 26, in select_edges
selected_edges=norm_edge_prop.flatten().multinomial(num_samples=n_FoF_links,replacement=False)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
RuntimeError: number of categories cannot exceed 2^24

During handling of the above exception, another exception occurred:

Traceback (most recent call last):
File "/gpfs/work1/0/einf4446/Pranav/DGL-PTM/DGL_testing/gpu_test.py", line 32, in
model.run()
File "/gpfs/work1/0/einf4446/Pranav/DGL-PTM/dgl_ptm/dgl_ptm/model/initialize_model.py", line 338, in run
self.step()
File "/gpfs/work1/0/einf4446/Pranav/DGL-PTM/dgl_ptm/dgl_ptm/model/initialize_model.py", line 332, in step
raise RuntimeError(f'execution of step failed for step {self.step_count}')
RuntimeError: execution of step failed for step 1