Add .data
.data
weight_data = net.module.layer1[0].conv1.weight.data print(weight_data.shape) print(weight_data.dtype)