Learning Invariances for Policy Generalization