mirrors/yolov5
1
0
Fork 0
mirror of https://github.com/ultralytics/yolov5.git synced 2025-06-03 14:49:29 +08:00
Code Issues Projects Releases Wiki Activity

reformat code

Browse Source
This commit is contained in:
Glenn Jocher 2020-08-14 11:53:44 -07:00
parent b35f76e3d0
commit f5da528d28
5 changed files with 15 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
utils/torch_utils.py
View File

@ -12,6 +12,7 @@ import torchvision.models as models
logger = logging.getLogger(__name__) logger = logging.getLogger(__name__)
def init_seeds(seed=0): def init_seeds(seed=0):
torch.manual_seed(seed) torch.manual_seed(seed)
@ -144,7 +145,8 @@ def model_info(model, verbose=False):
except: except:
fs = '' fs = ''
logger.info('Model Summary: %g layers, %g parameters, %g gradients%s' % (len(list(model.parameters())), n_p, n_g, fs)) logger.info(
'Model Summary: %g layers, %g parameters, %g gradients%s' % (len(list(model.parameters())), n_p, n_g, fs))
def load_classifier(name='resnet101', n=2): def load_classifier(name='resnet101', n=2):