Diff of /models/BloodNet3D.py [000000] .. [249e74]

Switch to unified view

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