TypeError: forward() missing 4 required positional arguments: 'prediction', 'instances', 'labels', and 'center_images'[BUG]

[kenta-takizawa]

Hello. I tried tutorial of bbbc010-2012 Jupyter notebooks, but this error happend and I don't know solution. Could you tell me what I should do ?

I ran 01-data.ipynb and 02-train.ipynb. When I ran 「begin_training(train_dataset_dict, val_dataset_dict, model_dict, loss_dict, configs, color_map=new_cmap)」, the following error happend.

Environment

• OS: Ubuntu 18.04
• GPU:Tesla
  -python3.7
  torch 1.1.0 torchvision 0.3.0 cuda=10.0

[lmanan]

Hello @kenta-takizawa,
Thanks for looking into EmbedSeg!
Could you also please share the 01-data and 02-train notebooks which you ran as well (for example, by exporting them to html. This can be done by selecting File>Download as> html)? Thanks!

[kenta-takizawa]

Thank you for your response. I uploaded jupyter notebooks in the following link. Please confirm.
https://github.com/kenta-takizawa/EmbedSeg_try