This commit is contained in:
henryruhs
2025-03-05 17:08:07 +01:00
parent aedaa20d78
commit 6388727262
+1
View File
@@ -3,6 +3,7 @@ from typing import Tuple
import torch
from torch import Tensor, nn
class UNet(nn.Module):
def __init__(self, output_size : int) -> None:
super().__init__()