You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Copy file name to clipboardExpand all lines: models/__init__.py
+2-2Lines changed: 2 additions & 2 deletions
Original file line number
Diff line number
Diff line change
@@ -10,8 +10,8 @@
10
10
11
11
In the function <__init__>, you need to define four lists:
12
12
-- self.loss_names (str list): specify the training losses that you want to plot and save.
13
-
-- self.model_names (str list): specify the images that you want to display and save.
14
-
-- self.visual_names (str list): define networks used in our training.
13
+
-- self.model_names (str list): define networks used in our training.
14
+
-- self.visual_names (str list): specify the images that you want to display and save.
15
15
-- self.optimizers (optimizer list): define and initialize optimizers. You can define one optimizer for each network. If two networks are updated at the same time, you can use itertools.chain to group them. See cycle_gan_model.py for an usage.
16
16
17
17
Now you can use the model class by specifying flag '--model dummy'.
0 commit comments