Skip to content

Commit

Permalink
added collate_fn import to example
Browse files Browse the repository at this point in the history
  • Loading branch information
conradry committed Feb 10, 2021
1 parent f706695 commit bb8b691
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -29,6 +29,7 @@ MaX-DeepLab has a complex architecture, training procedure and loss function. An
from max_deeplab.model import MaXDeepLabS
from max_deeplab.losses import MaXDeepLabLoss
from datasets.coco_panoptic import build
from util.misc import collate_fn

config = {}
config['image_size'] = (640, 640)
Expand Down

0 comments on commit bb8b691

Please sign in to comment.