Как применить групповую нормализацию к обучению PyTorch FasterRCNN? - PullRequest
0 голосов
/ 19 октября 2019

Я слежу за этой реализацией RCNN (вместо MaskRCNN я буду использовать FasterRCNN) https://pytorch.org/tutorials/intermediate/torchvision_tutorial.html.

Я хочу использовать нормализацию группы для входного слоя в моем классе DataSet. Я не хочу использовать пользовательский конвейер и создавать свои DataSet и DataLoader, так как обучающий конвейер PyTorch очень удобен в использовании.

Возможно ли использовать GroupNormalization со стандартным Dataloader и Dataset из torch.utils.data?

Существует один (я думаю) доступный DataLoader: -> https://pytorch.org/docs/stable/data.html#torch.utils.data.DataLoader и пара DataSets: -> Набор данных: https://pytorch.org/docs/stable/data.html#torch.utils.data.Dataset -> IterableDataset: https://pytorch.org/docs/stable/data.html#torch.utils.data.IterableDataset ...

Спасибо!

...