2019-01-09から1日間の記事一覧

PyTorchでTorch Hubに自作モデルの登録

PyTorchの自作モデルをTorch Hubに登録してみる. GitHub repo hubconf.pyの設置 以下のようにhubconf.pyをレポジトリの直下に設置する. # -*- coding: utf-8 -*- from models import MobileNet_v2 def mobilenet_v2(pretrained=False, *args, **kwargs): m…