monai registration notebook is not updated with latest training plan structure (and has a bug)

The monai registration notebook is not up to date with the latest training plan structure.

Furthermore, when I execute it it fails with the following error

2022-09-26 10:37:03,271 - {"asctime": "2022-09-26 10:37:03,271", "node_id": "node_d5ddbd94-53a0-4bc6-9d39-1727d5dba27d", "name": "fedbiomed", "level": "ERROR", "message": "Cannot train model in round: Boolean value of Tensor with more than one value is ambiguous"}