2021-08-01から1ヶ月間の記事一覧
何もしていない 寝る
弓と禅を読んだ。 他何もせず とりあえずMbnetv2からはじめて書くからTorchで書いている。早い段階でJaxで書きたい。 これの写経、明日には書けるはず github.comtimm はこのあたりはclass EfficientNetBuilderの部分で実装しているがわかりづらいね。 (Effi…
したこと Jaxでvggモデル実装 from functools import partial from typing import Callable, Optional, Sequence, Tuple, List, Union import jax.numpy as jnp from flax import linen as nn from .common import ConvBlock, ModuleDef, Sequential ModuleD…
プログラミング 以下の写経 github.com class CNN(nn.Module): @nn.compact def __call__(self, x): x = nn.Conv(32, kernel=(3, 3))(x) x = nn.relu(x) x = nn.avg_pool(x, window_shape=(2, 2), strides=(2, 2)) x = nn.Conv(64, kernel=(3, 3))(x) x = nn…
ブログでプログラミングのことは発信していましたが簡潔に自分用メモを残したいということで新たに日記も始めようと思います。 日記は毎日更新予定です。