Toggle navigation
Toggle navigation
This project
Loading...
Sign in
2020-2-capstone-design2
/
2015104192
Go to a project
Toggle navigation
Toggle navigation pinning
Projects
Groups
Snippets
Help
Project
Activity
Repository
Pipelines
Graphs
Issues
0
Merge Requests
0
Wiki
Snippets
Network
Create a new issue
Builds
Commits
Issue Boards
Authored by
윤영빈
2020-10-17 21:05:11 +0900
Browse Files
Options
Browse Files
Download
Email Patches
Plain Diff
Commit
a61db5586e7c75e4e81e865da569d0c280206575
a61db558
1 parent
e3b6e3b4
면담보고서, model parameter adjusted
Hide whitespace changes
Inline
Side-by-side
Showing
3 changed files
with
7 additions
and
7 deletions
web/backend/yt8m/frame_level_models.py
web/backend/yt8m/train.py
면담확인서/캡스톤 디자인 2 면담확인서 201016.docx
web/backend/yt8m/frame_level_models.py
View file @
a61db55
...
...
@@ -407,7 +407,7 @@ class NetVLADModelLF(models.BaseModel):
random_frames
=
True
cluster_size
=
64
hidden1_size
=
1024
relu
=
Fals
e
relu
=
Tru
e
dimred
=
-
1
gating
=
True
remove_diag
=
False
...
...
web/backend/yt8m/train.py
View file @
a61db55
...
...
@@ -75,7 +75,7 @@ if __name__ == "__main__":
flags
.
DEFINE_integer
(
"num_gpu"
,
1
,
"The maximum number of GPU devices to use for training. "
"Flag only applies if GPUs are installed"
)
flags
.
DEFINE_integer
(
"batch_size"
,
256
,
flags
.
DEFINE_integer
(
"batch_size"
,
128
,
"How many examples to process per batch for training."
)
flags
.
DEFINE_string
(
"label_loss"
,
"CrossEntropyLoss"
,
"Which loss function to use for training the model."
)
...
...
@@ -83,24 +83,24 @@ if __name__ == "__main__":
"regularization_penalty"
,
1.0
,
"How much weight to give to the regularization loss (the label loss has "
"a weight of 1)."
)
flags
.
DEFINE_float
(
"base_learning_rate"
,
0.0
1
,
flags
.
DEFINE_float
(
"base_learning_rate"
,
0.0
006
,
"Which learning rate to start with."
)
flags
.
DEFINE_float
(
"learning_rate_decay"
,
0.
95
,
"learning_rate_decay"
,
0.
8
,
"Learning rate decay factor to be applied every "
"learning_rate_decay_examples."
)
flags
.
DEFINE_float
(
"learning_rate_decay_examples"
,
40000
00
,
"learning_rate_decay_examples"
,
1
00
,
"Multiply current learning rate by learning_rate_decay "
"every learning_rate_decay_examples."
)
flags
.
DEFINE_integer
(
"num_epochs"
,
100
,
"How many passes to make over the dataset before "
"num_epochs"
,
5
,
"How many passes to make over the dataset before "
"halting training."
)
flags
.
DEFINE_integer
(
"max_steps"
,
None
,
"The maximum number of iterations of the training loop."
)
flags
.
DEFINE_integer
(
"export_model_steps"
,
1
,
"export_model_steps"
,
1
00
,
"The period, in number of steps, with which the model "
"is exported for batch prediction."
)
...
...
면담확인서/캡스톤 디자인 2 면담확인서 201016.docx
0 → 100644
View file @
a61db55
No preview for this file type
Please
register
or
login
to post a comment