Skip to content
GitLab
Menu
Projects
Groups
Snippets
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in
Toggle navigation
Menu
Open sidebar
Andrei-Claudiu Roibu
BrainMapper
Commits
3c3b8f2a
Commit
3c3b8f2a
authored
Apr 15, 2020
by
Andrei-Claudiu Roibu
🖥
Browse files
moved LogWriter initialization before load_checkpoint constructor
parent
4b4ac4d4
Changes
1
Show whitespace changes
Inline
Side-by-side
solver.py
View file @
3c3b8f2a
...
...
@@ -108,15 +108,16 @@ class Solver():
self
.
best_mean_score
=
0
self
.
best_mean_score_epoch
=
0
if
use_last_checkpoint
:
self
.
load_checkpoint
()
self
.
LogWriter
=
LogWriter
(
number_of_classes
=
number_of_classes
,
logs_directory
=
logs_directory
,
experiment_name
=
experiment_name
,
use_last_checkpoint
=
use_last_checkpoint
,
labels
=
labels
)
if
use_last_checkpoint
:
self
.
load_checkpoint
()
def
train
(
self
,
train_loader
,
test_loader
):
"""Training Function
...
...
Write
Preview
Supports
Markdown
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