nn.Conv2d expects a 4-dimensional input tensor as [batch_size, channels, height, width], while you are flattening the input tensor via:
b_x = x.view(-1, 750*750).to(device)

Please, let’s stick to one topic, e.g. this one 405.