Skip to content
GitLab
Projects
Groups
Snippets
Help
Loading...
Help
What's new
7
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in
Toggle navigation
Open sidebar
sjjsmuel
keras-frcnn
Commits
447ca5ee
Commit
447ca5ee
authored
Mar 09, 2020
by
Jonas Müller
Browse files
Options
Browse Files
Download
Email Patches
Plain Diff
better output
parent
9bcf8e78
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
4 additions
and
2 deletions
+4
-2
train_frcnn.py
train_frcnn.py
+4
-2
No files found.
train_frcnn.py
View file @
447ca5ee
...
...
@@ -210,7 +210,8 @@ class_mapping_inv = {v: k for k, v in class_mapping.items()}
print
(
'Starting training'
)
vis
=
True
log
=
open
(
options
.
out_log
,
"w"
)
log
=
open
(
options
.
out_log
,
"a"
)
log
.
write
(
"NEW RUN"
)
for
epoch_num
in
range
(
num_epochs
):
progbar
=
None
...
...
@@ -312,7 +313,8 @@ for epoch_num in range(num_epochs):
print
(
out
)
log
.
write
(
"---
\n
Epoch "
+
str
(
epoch_num
)
+
"
\n
"
)
log
.
write
(
out
+
"
\n
"
)
log
.
flush
()
best_loss
=
curr_loss
best_loss_epoch_num
=
epoch_num
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a 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