a | b/models/BloodNet3D.py | ||
---|---|---|---|
1 | import torch |
||
2 | import torch.nn as nn |
||
3 | |||
4 | class BloodNet3D(nn.Module): |
||
5 | def __init__(self): |
||
6 | super(BloodNet3D, self).__init__() |
||
7 | self.conv1 = nn.Conv3d(1, 32, kernel_size=3, padding=1) |
||
8 | # ... Add more layers as needed |
||
9 | |||
10 | def forward(self, x): |
||
11 | # Define the forward pass |
||
12 | return x |