Skip to content

Instantly share code, notes, and snippets.

@david-littlefield
Last active November 6, 2020 20:08
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save david-littlefield/0e3d409acb4648936a3404d784e7a906 to your computer and use it in GitHub Desktop.
Save david-littlefield/0e3d409acb4648936a3404d784e7a906 to your computer and use it in GitHub Desktop.
# Line 191
def init_priors(self ,cfg , min_size=cfg['min_sizes'], max_size=cfg['max_sizes']):
priorbox = PriorBox(cfg , min_size, max_size)
with torch.no_grad(): # <----------
prior = Variable( priorbox.forward()) # <----------
return prior
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment