search

novice03 / mobiledet-pytorch

PyTorch Implementation of MobileDet (https://arxiv.org/abs/2004.14525v3) backbones.
11 stars 6 forks source link

Add missing imports and requirements file #5

Closed dmckinnon closed 1 week ago

dmckinnon commented 8 months ago

Original code does not import torch.nn Requirements file also for easy pip install of reqs