Error when starts training Center Mask

youngwanLEE / centermask2

[CVPR 2020] CenterMask : Real-time Anchor-Free Instance Segmentation

Other

778 stars 159 forks source link

Closed haderalim closed 3 years ago

haderalim commented 3 years ago

Screen Shot 2020-11-16 at 12 00 11 PM

In train_net.py:-

The init function in "DefaultTrainer" takes 1 positional argument, but in train_net.py... 4 were given.
If I change super(DefaultTrainer, self).init(model, data_loader, optimizer) to super(DefaultTrainer, self).init(cfg) it is gives another error. So how can I fix this problem?

haderalim commented 3 years ago

@youngwanLEE

Peter-weng commented 3 years ago

just delete super(DefaultTrainer, self).init(model, data_loader, optimizer) and use it super().init(cfg) that maybe ok

youngwanLEE commented 3 years ago

@haderalim @Peter-weng update the code. Please re-open this issue if the problem continues.