Hi, if you read the error, it infers the input as a tensor. You need to explicitly specify it’s a dict with annotation
from typing import Dict
import torch

def forward_1(self, target_object_inputs_dict: Dict[str, torch.Tensor]):