maubin/ML_GravPotBCs
0
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

expansion

Browse source
This commit is contained in:
Mayeul Aubin 2025-06-06 13:52:17 +02:00
parent 24c2d546db
commit c07ec8f8cf
6 changed files with 138 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
sCOCA_ML/models/UNet_models.py
View file

@ -49,7 +49,7 @@ class UNet3D(BaseModel):
The FiLM layers are used to condition the feature maps on style parameters.
"""
super().init(N=N,
super().__init__(N=N,
in_channels=in_channels,
out_channels=out_channels,
style_parameters=style_dim,