Problem when running the HOW TO MAP FLOODWATER FROM RADAR IMAGERY USING SEMANTIC SEGMENTATION - BENCHMARK tutorial

Hello I am trying to run the tutorial as it is but I get this error message:

AttributeError: ‘functools.partial’ object has no attribute 'name

I am using python 3.9.6

Thanks in advance

Fabiano