@Jelena_Jevremovic, this is an issue where the grader has an older version of PyTorch and the graders version of torch.cat can’t handle different types, so you need to make sure you need to make sure both values you pass to it are floats.
1 Like