mirror of
https://github.com/lllyasviel/stable-diffusion-webui-forge.git
synced 2026-02-23 00:03:57 +00:00
43 lines
885 B
Python
43 lines
885 B
Python
# 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)
|
|
|
|
|