qq_42820467的博客

def load_param(self, model_path): param_dict = torch.load(model_path)# =============================================================================# for i in param_dict:# if 'fc' in i:# continue#

RuntimeError: [enforce fail at …\c10\core\CPUAllocator.cpp:75] data. DefaultCPUAllocator: not enough memory: you tried to allocate 50331648 bytes. Buy new RAM!pytorch:RuntimeError: CUDA error: CUBLAS_STATUS_ALLOC_FAILED when calling cublasCreate(handle)b