tf_mdl_dir = 'data/20180402-114759'
data_name = 'vggface2'
state_dict = mdl.state_dict()
torch.save(state_dict, f'{tf_mdl_dir}-{data_name}.pt')
# VS
model_name = '{}-{}.pt'.format(tf_mdl_dir, data_name)
#PS: 还是后一种比较简明易懂
{python} {字符串格式}
猜你喜欢
转载自blog.csdn.net/tony2278/article/details/103887574
今日推荐
周排行