pytorch-layoutnet
pytorch-layoutnet copied to clipboard
how many GB RAM Is required for this???? i am getting this error
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
For panoContext_img_train.t7, it will load 414 x 512 x 1024 x 3 bytes.