Создать аналогичную голову в Керасе - PullRequest
Новая
       9

Создать аналогичную голову в Керасе

0 голосов
/ 04 марта 2020

Когда я создаю CNN в Быстром ИИ, используя трансферное обучение, создается голова, подобная этой: 

(1): Sequential(
    (0): AdaptiveConcatPool2d(
      (ap): AdaptiveAvgPool2d(output_size=1)
      (mp): AdaptiveMaxPool2d(output_size=1)
    )
    (1): Flatten()
    (2): BatchNorm1d(3840, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)
    (3): Dropout(p=0.05005, inplace=False)
    (4): Linear(in_features=3840, out_features=512, bias=True)
    (5): ReLU(inplace=True)
    (6): BatchNorm1d(512, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)
    (7): Dropout(p=0.1001, inplace=False)
    (8): Linear(in_features=512, out_features=3, bias=True)

Я хотел бы создать ее как можно ближе к этой в Керасе, однако я застрял на AdaptiveConcatPool2D часть этого. Похоже, в настоящее время нет классов, подобных этому. Любые идеи о том, как воспроизвести это как можно ближе в Керасе?

...