Skip to content

Commit

Permalink
fix model arch
Browse files Browse the repository at this point in the history
  • Loading branch information
ziw-liu committed Aug 10, 2023
1 parent 998bfde commit b81fd0e
Showing 1 changed file with 2 additions and 0 deletions.
2 changes: 2 additions & 0 deletions examples/demo_dlmbl/python/excercise_1.py
Original file line number Diff line number Diff line change
Expand Up @@ -177,6 +177,8 @@
# %%
model_config = {
"architecture": "2D",
"in_channels": 1,
"out_channels": 2,
"residual": True,
"dropout": 0.1,
"task": "reg",
Expand Down

0 comments on commit b81fd0e

Please sign in to comment.