I m running into an error:
RuntimeError: Input type (torch.FloatTensor) and weight type (torch.cuda.FloatTensor) should be the same or input should be a MKLDNN tensor and weight is a dense tensor`
When I shift it to the CPU:
hl.build_graph(self.online_net.to('cpu'), torch.zeros([1, 4, 32, 32]))
I m getting
RuntimeError: Expected node type 'onnx::Constant' for argument 'rounding_mode' of node '_div_rounding_mode', got 'prim::Param'.
I will also print my code. Maybe somebody can figure out why its not working.
Hello thank you for sharing your lib.
I m running into an error:
RuntimeError:
Input type (torch.FloatTensor) and weight type (torch.cuda.FloatTensor) should be the same or input should be a MKLDNN tensor and weight is a dense tensor`When I shift it to the CPU:
hl.build_graph(self.online_net.to('cpu'), torch.zeros([1, 4, 32, 32]))
I m getting
RuntimeError: Expected node type 'onnx::Constant' for argument 'rounding_mode' of node '_div_rounding_mode', got 'prim::Param'.
I will also print my code. Maybe somebody can figure out why its not working.