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.