(format, ev_dir, log_suffix="")
| 189 | |
| 190 | |
| 191 | def make_output_format(format, ev_dir, log_suffix=""): |
| 192 | os.makedirs(ev_dir, exist_ok=True) |
| 193 | if format == "stdout": |
| 194 | return HumanOutputFormat(sys.stdout) |
| 195 | elif format == "log": |
| 196 | return HumanOutputFormat(osp.join(ev_dir, "log%s.txt" % log_suffix)) |
| 197 | elif format == "json": |
| 198 | return JSONOutputFormat(osp.join(ev_dir, "progress%s.json" % log_suffix)) |
| 199 | elif format == "csv": |
| 200 | return CSVOutputFormat(osp.join(ev_dir, "progress%s.csv" % log_suffix)) |
| 201 | elif format == "tensorboard": |
| 202 | return TensorBoardOutputFormat(osp.join(ev_dir, "tb%s" % log_suffix)) |
| 203 | else: |
| 204 | raise ValueError("Unknown format specified: %s" % (format,)) |
| 205 | |
| 206 | |
| 207 | # ================================================================ |
no test coverage detected