| 223 | optimizer.load_state_dict(state) |
| 224 | |
| 225 | def create_logger(name, log_file, level=logging.INFO): |
| 226 | l = logging.getLogger(name) |
| 227 | formatter = logging.Formatter('[%(asctime)s][%(filename)20s][line:%(lineno)4d][%(levelname)8s] %(message)s') |
| 228 | fh = logging.FileHandler(log_file) |
| 229 | fh.setFormatter(formatter) |
| 230 | sh = logging.StreamHandler() |
| 231 | sh.setFormatter(formatter) |
| 232 | l.setLevel(level) |
| 233 | l.addHandler(fh) |
| 234 | l.addHandler(sh) |
| 235 | return l |
| 236 | |
| 237 | class IterLRScheduler(object): |
| 238 | def __init__(self, optimizer, milestones, lr_mults, last_iter=-1): |