andras/bipolaroid
Watch
1
0
Fork
You've already forked bipolaroid
 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Update networks

Browse source
This commit is contained in:
Andras Schmelczer 2024-06-22 15:01:02 +01:00
commit b7755bfea8
No known key found for this signature in database
GPG key ID: FC8F2C3D3D1A718C
1 changed files with 173 additions and 144 deletions
Download patch file Download diff file Expand all files Collapse all files

4
src/editor/models/simple_cnn.py
View file

@ -3,10 +3,8 @@ import torch.nn.functional as F
class SimpleCNN(nn.Module):
def __init__(self, bin_count):
def __init__(self, **_):
super(SimpleCNN, self).__init__()
self.bin_count = bin_count
# Define the convolutional layers
self.conv1 = nn.Conv3d(
1, 16, kernel_size=3, padding=1