Skip to content

Commit

Permalink
style: 🎨 run black
Browse files Browse the repository at this point in the history
  • Loading branch information
neptunes5thmoon committed Aug 23, 2024
1 parent bcdf00c commit 2628275
Showing 1 changed file with 5 additions and 1 deletion.
6 changes: 5 additions & 1 deletion solution.py
Original file line number Diff line number Diff line change
Expand Up @@ -1191,7 +1191,11 @@ def launch_tensorboard(log_dir):

# %% tags=["solution"]
model = UNet(
depth=3, in_channels=1, upsample_mode="bilinear", fmap_inc_factor=3, final_activation=torch.nn.Sigmoid()
depth=3,
in_channels=1,
upsample_mode="bilinear",
fmap_inc_factor=3,
final_activation=torch.nn.Sigmoid(),
) # SOLUTION 8.1: Declare your U-Net here and name it below
model_name = "my_fav_unet" # This name will be used in the tensorboard logs
logger = SummaryWriter(f"unet_runs/{model_name}")
Expand Down

0 comments on commit 2628275

Please sign in to comment.