pytorch-layoutnet icon indicating copy to clipboard operation
pytorch-layoutnet copied to clipboard

how many GB RAM Is required for this???? i am getting this error

Open anish9 opened this issue 6 years ago • 1 comments

panoContext_img_train.t7      : 414 examples
Traceback (most recent call last):
  File "torch2pytorch_data.py", line 83, in <module>
    cvt2png(os.path.join(DATA_DIR, 'train'), train_pats, train_pano_map)
  File "torch2pytorch_data.py", line 65, in cvt2png
    imgs = load_lua(th_path).numpy()
  File "/opt/anaconda3/lib/python3.6/site-packages/torch/utils/serialization/read_lua_file.py", line 608, in load_lua
    return reader.read()
  File "/opt/anaconda3/lib/python3.6/site-packages/torch/utils/serialization/read_lua_file.py", line 593, in read
    return self.read_object()
  File "/opt/anaconda3/lib/python3.6/site-packages/torch/utils/serialization/read_lua_file.py", line 523, in wrapper
    result = fn(self, *args, **kwargs)
  File "/opt/anaconda3/lib/python3.6/site-packages/torch/utils/serialization/read_lua_file.py", line 546, in read_object
    return reader_registry[cls_name](self, version)
  File "/opt/anaconda3/lib/python3.6/site-packages/torch/utils/serialization/read_lua_file.py", line 147, in read_tensor
    storage = reader.read()
  File "/opt/anaconda3/lib/python3.6/site-packages/torch/utils/serialization/read_lua_file.py", line 593, in read
    return self.read_object()
  File "/opt/anaconda3/lib/python3.6/site-packages/torch/utils/serialization/read_lua_file.py", line 523, in wrapper
    result = fn(self, *args, **kwargs)
  File "/opt/anaconda3/lib/python3.6/site-packages/torch/utils/serialization/read_lua_file.py", line 546, in read_object
    return reader_registry[cls_name](self, version)
  File "/opt/anaconda3/lib/python3.6/site-packages/torch/utils/serialization/read_lua_file.py", line 166, in read_storage
    return python_class.from_buffer(reader.f.read(size), 'native')
RuntimeError: $ Torch: not enough memory: you tried to allocate 4GB. Buy new RAM! at /pytorch/aten/src/TH/THGeneral.cpp:204

anish9 avatar Oct 27 '18 14:10 anish9

For panoContext_img_train.t7, it will load 414 x 512 x 1024 x 3 bytes.

sunset1995 avatar Oct 28 '18 07:10 sunset1995