fix(utils): get_scheduler returns exception instead of (+3 more)

[andrewwhitecdw]

Small fixes in utils.py:

fix: get_scheduler returns exception instead of raising

Fix: Replace:

    else:
        return NotImplementedError('learning rate policy [%s] is not implemented', hyperparameters['lr_policy'])

with:

    else:
        raise NotImplementedError('learning rate policy [%s] is not implemented' % hyperparameters['lr_policy'])

fix: load_vgg16 uses load_lua and Vgg16 without imports

Fix: Replace:

import torch.nn.init as init
import time

with:

import torch.nn.init as init
import time
from torch.utils.serialization import load_lua
from networks import Vgg16

fix: get_model_list crashes when no matching checkpoints

Fix: Replace:

    if gen_models is None:
        return None
    gen_models.sort()

with:

    if not gen_models:
        return None
    gen_models.sort()

fix: load_vgg16 uses undefined load_lua and Vgg16 symbols

Fix: Replace:

import time
# Methods

with:

import time
try:
    from torch.utils.serialization import load_lua
except ImportError:
    load_lua = None
from vgg import Vgg16
# Methods

Files changed

• utils.py