Skip to content

Instantly share code, notes, and snippets.

@arunm8489

arunm8489/13

Created June 3, 2020 15:20
Show Gist options
  • Save arunm8489/ebb7958836620d7e43eccd0b026d3b58 to your computer and use it in GitHub Desktop.
Save arunm8489/ebb7958836620d7e43eccd0b026d3b58 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
13
class Darknet(nn.Module):
def __init__(self, cfgfile):
super(Darknet, self).__init__()
self.blocks = parse_cfg(cfgfile)
self.net_info, self.module_list = model_initialization(self.blocks)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment