Skip to content
GitLab
Projects
Groups
Snippets
/
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in
Toggle navigation
Menu
Open sidebar
Lucas Miranda
deepOF
Commits
9adbeaf5
Commit
9adbeaf5
authored
Apr 27, 2021
by
lucas_miranda
Browse files
Added a MirroredStrategy to train models on multiple GPUs if they are available
parent
4e2ee1c2
Pipeline
#100342
passed with stages
in 24 minutes and 46 seconds
Changes
1
Pipelines
1
Hide whitespace changes
Inline
Side-by-side
deepof/train_utils.py
View file @
9adbeaf5
...
...
@@ -443,7 +443,7 @@ def autoencoder_fitting(
+
[
CustomStopper
(
monitor
=
"val_loss"
,
patience
=
1
0
,
patience
=
1
5
,
restore_best_weights
=
True
,
start_epoch
=
max
(
kl_warmup
,
mmd_warmup
),
),
...
...
@@ -467,7 +467,7 @@ def autoencoder_fitting(
callbacks_
=
cbacks
+
[
CustomStopper
(
monitor
=
"val_loss"
,
patience
=
5
,
patience
=
1
5
,
restore_best_weights
=
True
,
start_epoch
=
max
(
kl_warmup
,
mmd_warmup
),
),
...
...
Write
Preview
Supports
Markdown
0%
Try again
or
attach a new file
.
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment