Unverified Commit 9680e09d authored by srihari-humbarwadi's avatar srihari-humbarwadi
Browse files

clean up training script

parent fb4b9580
......@@ -12,12 +12,8 @@
# See the License for the specific language governing permissions and
# limitations under the License.
"""Panoptic MaskRCNN trainer.
"""Panoptic MaskRCNN trainer."""
All custom registry are imported from registry_imports. Here we use default
trainer so we directly call train.main. If you need to customize the trainer,
branch from `official/vision/beta/train.py` and make changes.
"""
from absl import app
from official.common import flags as tfm_flags
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment