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

PyTorchでTorch Hubを使ったpre-trained modelの共有

Torch Hubは研究の再現性のために,PyTorchのpre-trained modelの共有のための機能. https://pytorch.org/docs/master/hub.html hubconf.py 以下のフォーマットでentrypointを定義する entrypointはPyTorchのモデルを返す関数 def entrypoint_name(pretrain…