AYO/dl_bench
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Add TF1 platform, docker and README

Browse source
This commit is contained in:
Corentin 2021-10-05 10:51:18 +09:00
commit dbe5490c5b
28 changed files with 655 additions and 34 deletions
Download patch file Download diff file Expand all files Collapse all files

10
config/benchmark.py
View file

@ -1,5 +1,15 @@
from enum import Enum
class Config:
class ExperimentCategory(Enum):
SMALL = [0, 20] # in Mop/experiment
MEDIUM = [20, 1000]
LARGE = [1000, 1_000_000]
VERY_LARGE = [1_000_000, 1_000_000_000]
EXPERIMENT_TIME = 1.0
ELEMENT_WISE_ARGS = [
(100, 100),
(100, 200),