本文整理汇总了Python中utils.mkdir方法的典型用法代码示例。如果您正苦于以下问题:Python utils.mkdir方法的具体用法?Python utils.mkdir怎么用?Python utils.mkdir使用的例子?那么恭喜您, 这里精选的方法代码示例或许可以为您提供帮助。您也可以进一步了解该方法所在模块utils的用法示例。
在下文中一共展示了utils.mkdir方法的14个代码示例,这些例子默认根据受欢迎程度排序。您可以为喜欢或者感觉有用的代码点赞,您的评价将有助于我们的系统推荐出更棒的Python代码示例。
示例1: snapshot
点赞 6
# 需要导入模块: import utils [as 别名]
# 或者: from utils import mkdir [as 别名]
def snapshot(self, path):
if not os.path.exists(path):
if os.name == 'nt':
os.system('mkdir {}'.format(path.replace('/', '\\')))
else:
os.system('mkdir -p {}'.format(path))
# save every 100 tick if the network is in stab phase.
ndis = 'dis_R{}_T{}.pth.tar'.format(int(floor(self.resl)), self.globalTick)
ngen = 'gen_R{}_T{}.pth.tar'.format(int(floor(self.resl)), self.globalTick)
if self.globalTick%50==0:
if self.phase == 'gstab' or self.phase =='dstab' or self.phase == 'final':
save_path = os.path.join(path, ndis)
if not os.path.exists(save_path):
torch.save(self.get_state('dis'), save_path)
save_path = os.path.join(path, ngen)
torch.save(self.get_state('gen'), save_path)
print('[snapshot] model saved @ {}'.format(path))
开发者ID:nashory,项目名称:pggan-pytorch,代码行数:19,
示例2: save_results
点赞 5
# 需要导入模块: import utils [as 别名]
# 或者: from utils import mkdir [as 别名]
def save_results(state, steps, visualize=True, subfolder=''):
if not state.get_output_flag():
logging.warning('Skip saving results because output_flag is False')
return
expr_dir = os.path.join(state.get_save_directory(), subfolder)
utils.mkdir(expr_dir)
save_data_path = os.path.join(expr_dir, 'results.pth')
steps = [(d.detach().cpu(), l.detach().cpu(), lr) for (d, l, lr) in steps]
if visualize:
vis_results(state, steps, expr_dir)
torch.save(steps, save_data_path)
logging.info('Results saved to {}'.format(save_data_path))
开发者ID:SsnL,项目名称:dataset-distillation,代码行数:17,代码来源:io.py
示例3: save_test_results
点赞 5
# 需要导入模块: import utils [as 别名]
# 或者: from utils import mkdir [as 别名]
def save_test_results(state, results):
assert state.phase != 'train'
if not state.get_output_flag():
logging.warning('Skip saving test results because output_flag is False')
return
test_dir = state.get_save_directory()
utils.mkdir(test_dir)
result_file = os.path.join(test_dir, 'results.pth')
torch.save(results, os.path.join(test_dir, 'results.pth'))
logging.info('Test results saved as {}'.format(result_file))
开发者ID:SsnL,项目名称:dataset-distillation,代码行数:13,代码来源:io.py
示例4: test_workteamjob
点赞 5
# 需要导入模块: import utils [as 别名]
# 或者: from utils import mkdir [as 别名]
def test_workteamjob(
kfp_client, experiment_id, region, sagemaker_client, test_file_dir
):
download_dir = utils.mkdir(os.path.join(test_file_dir + "/generated"))
workteam_name, workflow_json = create_workteamjob(
kfp_client, experiment_id, region, sagemaker_client, test_file_dir, download_dir
)
outputs = {"sagemaker-private-workforce": ["workteam_arn"]}
try:
output_files = minio_utils.artifact_download_iterator(
workflow_json, outputs, download_dir
)
response = sagemaker_utils.describe_workteam(sagemaker_client, workteam_name)
# Verify WorkTeam was created in SageMaker
assert response["Workteam"]["CreateDate"] is not None
assert response["Workteam"]["WorkteamName"] == workteam_name
# Verify WorkTeam arn artifact was created in Minio and matches the one in SageMaker
workteam_arn = utils.read_from_file_in_tar(
output_files["sagemaker-private-workforce"]["workteam_arn"],
"workteam_arn.txt",</