/usr/local/lib64/python3.6/site-packages/caffe2/contrib/playground
Edit: /usr/local/lib64/python3.6/site-packages/caffe2/contrib/playground/output_generator.py (478B)
from caffe2.python import timeout_guard
def fun_conclude_operator(self):
# Ensure the program exists. This is to "fix" some unknown problems
# causing the job sometimes get stuck.
timeout_guard.EuthanizeIfNecessary(600.0)
def assembleAllOutputs(self):
output = {}
output['train_model'] = self.train_model
output['test_model'] = self.test_model
output['model'] = self.model_output
output['metrics'] = self.metrics_output
return output