Spaces:
Running
on
Zero
Running
on
Zero
| # A reimplemented version in public environments by Xiao Fu and Mu Hu | |
| import json | |
| import yaml | |
| import logging | |
| import os | |
| import numpy as np | |
| import sys | |
| def load_loss_scheme(loss_config): | |
| with open(loss_config, 'r') as f: | |
| loss_json = yaml.safe_load(f) | |
| return loss_json | |
| DEBUG =0 | |
| logger = logging.getLogger() | |
| if DEBUG: | |
| #coloredlogs.install(level='DEBUG') | |
| logger.setLevel(logging.DEBUG) | |
| else: | |
| #coloredlogs.install(level='INFO') | |
| logger.setLevel(logging.INFO) | |
| strhdlr = logging.StreamHandler() | |
| logger.addHandler(strhdlr) | |
| formatter = logging.Formatter('%(asctime)s [%(filename)s:%(lineno)d] %(levelname)s %(message)s') | |
| strhdlr.setFormatter(formatter) | |
| def count_parameters(model): | |
| return sum(p.numel() for p in model.parameters() if p.requires_grad) | |
| def check_path(path): | |
| if not os.path.exists(path): | |
| os.makedirs(path, exist_ok=True) | |