os.chdir('result')
# d = datetime.datetime.today()
# dir_name = d.strftime('%Y-%m-%d_%H-%M-%S')
# print dir_name
# if not os.path.exists(dir_name):
# os.mkdir(dir_name)
# os.chdir(dir_name)
dir_name = 'finetune_epoch' + str(epoch)
if not os.path.exists(dir_name):
os.mkdir(dir_name)
os.chdir(dir_name)
# Feature
output = rbm1.output_from_input(rbm1.input)
output = rbm2.output_from_input(output)
output = rbm3.output_from_input(output)
output = rbm4.output_from_input(output)
output = rbm5.output_from_input(output)
output = rbm6.output_from_input(output)
output = rbm7.output_from_input(output)
saveFeatures(output)
# RBM1 Save
reinput = rbm1.reconstruct_from_input(rbm1.input)
saveImage(reinput, node_shape[2], 'rbm1_after_train')
saveW(rbm1.getW(), 'rbm1_after_train')
# RBM2 Save
output = rbm1.output_from_input(rbm1.input)
