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
2952a4df
Commit
2952a4df
authored
Apr 27, 2021
by
lucas_miranda
Browse files
Added a MirroredStrategy to train models on multiple GPUs if they are available
parent
7e69c2f7
Pipeline
#100335
passed with stages
in 27 minutes and 8 seconds
Changes
1
Pipelines
1
Hide whitespace changes
Inline
Side-by-side
deepof/train_utils.py
View file @
2952a4df
...
...
@@ -443,7 +443,7 @@ def autoencoder_fitting(
+
[
CustomStopper
(
monitor
=
"val_loss"
,
patience
=
5
,
patience
=
10
,
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