53 lines
1.6 KiB
Python
53 lines
1.6 KiB
Python
# Copyright (c) Facebook, Inc. and its affiliates.
|
|
# All rights reserved.
|
|
#
|
|
# This source code is licensed under the license found in the
|
|
# LICENSE file in the root directory of this source tree.
|
|
#
|
|
import time, sys
|
|
import numpy as np
|
|
|
|
def time_for_file():
|
|
ISOTIMEFORMAT='%d-%h-at-%H-%M-%S'
|
|
return '{}'.format(time.strftime( ISOTIMEFORMAT, time.gmtime(time.time()) ))
|
|
|
|
def time_string():
|
|
ISOTIMEFORMAT='%Y-%m-%d %X'
|
|
string = '[{}]'.format(time.strftime( ISOTIMEFORMAT, time.gmtime(time.time()) ))
|
|
return string
|
|
|
|
def time_string_short():
|
|
ISOTIMEFORMAT='%Y%m%d'
|
|
string = '{}'.format(time.strftime( ISOTIMEFORMAT, time.gmtime(time.time()) ))
|
|
return string
|
|
|
|
def time_print(string, is_print=True):
|
|
if (is_print):
|
|
print('{} : {}'.format(time_string(), string))
|
|
|
|
def convert_size2str(torch_size):
|
|
dims = len(torch_size)
|
|
string = '['
|
|
for idim in range(dims):
|
|
string = string + ' {}'.format(torch_size[idim])
|
|
return string + ']'
|
|
|
|
def convert_secs2time(epoch_time, return_str=False):
|
|
need_hour = int(epoch_time / 3600)
|
|
need_mins = int((epoch_time - 3600*need_hour) / 60)
|
|
need_secs = int(epoch_time - 3600*need_hour - 60*need_mins)
|
|
if return_str:
|
|
str = '[{:02d}:{:02d}:{:02d}]'.format(need_hour, need_mins, need_secs)
|
|
return str
|
|
else:
|
|
return need_hour, need_mins, need_secs
|
|
|
|
def print_log(print_string, log):
|
|
#if isinstance(log, Logger): log.log('{:}'.format(print_string))
|
|
if hasattr(log, 'log'): log.log('{:}'.format(print_string))
|
|
else:
|
|
print("{:}".format(print_string))
|
|
if log is not None:
|
|
log.write('{:}\n'.format(print_string))
|
|
log.flush()
|