diff --git a/.gitignore b/.gitignore index cfa88408d0b4ac0e81c34937226768c96bc0117b..bf472b1e79a4e7d55c8b0c550cbcc4249630dd19 100644 --- a/.gitignore +++ b/.gitignore @@ -2,3 +2,4 @@ output .idea/ *~ +*.pyc diff --git a/examples/train.py b/examples/train.py index 487ce6fe13d2e760e9ec7dc29e5c38b6238d35b1..c91b2f45c73ed79e7c2ec181db05eaf50e819005 100644 --- a/examples/train.py +++ b/examples/train.py @@ -24,13 +24,19 @@ # See the License for the specific language governing permissions and # limitations under the License. +import os + import yaml -from .. trainer.factory import TrainerFactory +from eleps.trainer.factory import TrainerFactory if __name__ == "__main__": with open('ctr-dnn_train.yaml', 'r') as rb: global_config = yaml.load(rb.read()) + print global_config + + os.exit() + trainer = TrainerFactory.create(global_config) trainer.run() diff --git a/metrics/__init__.py b/metrics/__init__.py new file mode 100755 index 0000000000000000000000000000000000000000..abf198b97e6e818e1fbe59006f98492640bcee54 --- /dev/null +++ b/metrics/__init__.py @@ -0,0 +1,13 @@ +# Copyright (c) 2020 PaddlePaddle Authors. All Rights Reserved. +# +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License.