TypeError: forward() missing 4 required positional arguments: 'prediction', 'instances', 'labels', and 'center_images'[BUG]
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
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!
Thank you for your response. I uploaded jupyter notebooks in the following link. Please confirm. https://github.com/kenta-takizawa/EmbedSeg_try