RuntimeError: Error(s) in loading state_dict for NerfModel:
size mismatch for MLP_0.rgb_layer.weight: copying a param with shape torch.Size([256, 256]) from checkpoint, the shape in current model is torch.Size([48, 256]).
size mismatch for MLP_0.rgb_layer.bias: copying a param with shape torch.Size([256]) from checkpoint, the shape in current model is torch.Size([48]).
size mismatch for MLP_1.rgb_layer.weight: copying a param with shape torch.Size([256, 256]) from checkpoint, the shape in current model is torch.Size([48, 256]).
size mismatch for MLP_1.rgb_layer.bias: copying a param with shape torch.Size([256]) from checkpoint, the shape in current model is torch.Size([48]).
RuntimeError: Error(s) in loading state_dict for NerfModel: size mismatch for MLP_0.rgb_layer.weight: copying a param with shape torch.Size([256, 256]) from checkpoint, the shape in current model is torch.Size([48, 256]). size mismatch for MLP_0.rgb_layer.bias: copying a param with shape torch.Size([256]) from checkpoint, the shape in current model is torch.Size([48]). size mismatch for MLP_1.rgb_layer.weight: copying a param with shape torch.Size([256, 256]) from checkpoint, the shape in current model is torch.Size([48, 256]). size mismatch for MLP_1.rgb_layer.bias: copying a param with shape torch.Size([256]) from checkpoint, the shape in current model is torch.Size([48]).