mirror of
https://github.com/unanmed/ginka-generator.git
synced 2026-05-17 23:21:20 +08:00
14 lines
376 B
Python
14 lines
376 B
Python
import torch
|
|
import torch.nn as nn
|
|
|
|
class GinkaOutput(nn.Module):
|
|
def __init__(self, out_ch=32, base_ch=64, out_size=(13, 13)):
|
|
super().__init__()
|
|
self.conv_down = nn.Sequential(
|
|
nn.AdaptiveMaxPool2d(out_size),
|
|
nn.Conv2d(base_ch, out_ch, 1)
|
|
)
|
|
|
|
def forward(self, x):
|
|
return self.conv_down(x)
|