ristea / sspcab

Other
147 stars 15 forks source link

RuntimeError: [enforce fail at C:\actions-runner\_work\pytorch\pytorch\builder\windows\pytorch\c10\core\impl\alloc_cpu.cpp:72] data. DefaultCPUAllocator: not enough memory: you tried to allocate 274877906944 bytes. #11

Open wangxin-fighting opened 11 months ago

wangxin-fighting commented 11 months ago

Hello, I encountered the following error after replacing your code with my original class MLP(). Can you help me see how to solve it?

File "D:\1-study\2-code\FOD-best\models\model.py", line 83, in init self.conv1 = nn.Conv2d(in_channels=channels, File "D:\ProgramData\Anaconda3\envs\wx39\lib\site-packages\torch\nn\modules\conv.py", line 450, in init super(Conv2d, self).init( File "D:\ProgramData\Anaconda3\envs\wx39\lib\site-packages\torch\nn\modules\conv.py", line 137, in init self.weight = Parameter(torch.empty( RuntimeError: [enforce fail at C:\actions-runner_work\pytorch\pytorch\builder\windows\pytorch\c10\core\impl\alloc_cpu.cpp:72] data. DefaultCPUAllocator: not enough memory: you tried to allocate 274877906944 bytes.