Как получить доступ к имени папки перед сохранением файла pb в Tensorflow?

Question

Я сохранил pb-файл с помощью сценария Tensorflow, но похоже, что выходная папка, которую я указал, используется в качестве родительской папки, а Tensorflow создал подпапку с заголовком цифрой c (например, 1586312775), а затем сохранил файл pb внутри него.

estimator.export_savedmodel('/model_output', serving_input_receiver_fn=serving_input_receiver_fn())

И структура выходной папки выглядит следующим образом:

model_output -- 1586312775 -- variables
                           -- saved_model.pb

Существует ли какой-либо метод для доступа к автоматическому c имени папки (1586312775), которое созданный Tensorflow? Могу ли я изменить имя папки, прежде чем она сохранит файл pb под ней?

Спасибо,

Answer 1

def get_dist_folder_name(estimator):
    job_name = estimator.latest_training_job.job_name
    base_job_name = estimator.base_job_name
    job_timestamp = job_name[job_name.find(base_job_name)+len(base_job_name)+1:][:-4]
    target_folder_name = str(int(time.mktime(datetime.datetime.strptime(job_timestamp, "%Y-%m-%d-%H-%M-%S").timetuple())))
    return target_folder_name 

Кстати, я делал это на AWS, и это получит имя папки с меткой времени ....