DeepInf bug: MemoryError: Unable to allocate 7.26 GiB for an array with shape (779164, 50, 50) and d

Traceback (most recent call last):
  File "train.py", line 101, in 
    args.file_dir, args.dim, args.seed, args.shuffle, args.model)
  File "/root/DeepInf/data_loader.py", line 48, in __init__
    self.graphs = np.load(os.path.join(file_dir, "adjacency_matrix.npy")).astype(np.float32)
MemoryError: Unable to allocate 7.26 GiB for an array with shape (779164, 50, 50) and data type float32

你可能感兴趣的:(python)