From e986f2c731f35875ae2c02c344f5b7baa0ea50c7 Mon Sep 17 00:00:00 2001 From: Wendi Li Date: Mon, 23 Nov 2020 16:25:29 +0800 Subject: [PATCH] Add files via upload --- examples/benchmarks/TFT/README.md | 10 ++++++++++ examples/benchmarks/TFT/requirements.txt | 3 +++ 2 files changed, 13 insertions(+) create mode 100644 examples/benchmarks/TFT/README.md create mode 100644 examples/benchmarks/TFT/requirements.txt diff --git a/examples/benchmarks/TFT/README.md b/examples/benchmarks/TFT/README.md new file mode 100644 index 000000000..03d45e087 --- /dev/null +++ b/examples/benchmarks/TFT/README.md @@ -0,0 +1,10 @@ +# Temporal Fusion Transformers Benchmark +## Source +https://github.com/google-research/google-research/tree/master/tft + +## Run the Workflow +Users can follow the ``workflow_by_code_tft.py`` to run the benchmark. + +### Notes +1. The model must run in GPU, or an error will be raised. +2. New datasets should be registered in ``data_formatters``, for detail please visit the source. \ No newline at end of file diff --git a/examples/benchmarks/TFT/requirements.txt b/examples/benchmarks/TFT/requirements.txt new file mode 100644 index 000000000..04234aaed --- /dev/null +++ b/examples/benchmarks/TFT/requirements.txt @@ -0,0 +1,3 @@ +tensorflow-gpu==1.15.0 +numpy == 1.19.4 +pandas==1.1.0 \ No newline at end of file