"test/torchaudio_unittest/backend/utils_test.py" did not exist on "9ba02d5b0edcf5d20f2eec15a188a7669ae2f725"
train.py 1.1 KB
Newer Older
A. Unique TensorFlower's avatar
A. Unique TensorFlower committed
1
# Copyright 2022 The TensorFlow Authors. All Rights Reserved.
Gunho Park's avatar
Gunho Park committed
2
3
4
5
6
7
8
9
10
11
12
13
#
# 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.
Gunho Park's avatar
Gunho Park committed
14

15
"""TensorFlow Model Garden Vision training driver."""
Gunho Park's avatar
Gunho Park committed
16
17
18

from absl import app

19
# pylint: disable=unused-import
Gunho Park's avatar
Gunho Park committed
20
from official.common import flags as tfm_flags
21
22
23
24
from official.projects.basnet.configs import basnet as basnet_cfg
from official.projects.basnet.modeling import basnet_model
from official.projects.basnet.modeling import refunet
from official.projects.basnet.tasks import basnet as basenet_task
Gunho Park's avatar
Gunho Park committed
25
from official.vision.beta import train
Gunho Park's avatar
Gunho Park committed
26
27
28
29


if __name__ == '__main__':
  tfm_flags.define_flags()
Gunho Park's avatar
Gunho Park committed
30
  app.run(train.main)