def METHOD_NAME(self, obs_dict):
    # we always work with dictionary observations. Primary observation is available with the key 'obs'
    main_obs = obs_dict["obs"]
    x = self.conv_head(main_obs)
    x = x.view(-1, self.conv_head_out_size)
    return x