com
.
intel
.
analytics
.
bigdl
.
models
.
resnet
.
Utils
TrainParams
case class
TrainParams
(
folder:
String
=
"./"
,
checkpoint:
Option
[
String
] =
scala.None
,
modelSnapshot:
Option
[
String
] =
scala.None
,
stateSnapshot:
Option
[
String
] =
scala.None
,
optnet:
Boolean
=
false
,
depth:
Int
=
20
,
classes:
Int
=
10
,
shortcutType:
String
=
"A"
,
batchSize:
Int
=
128
,
nepochs:
Int
=
165
,
learningRate:
Double
=
0.1
,
weightDecay:
Double
=
1.0E-4
,
momentum:
Double
=
0.9
,
dampening:
Double
=
0.0
,
nesterov:
Boolean
=
true
,
graphModel:
Boolean
=
false
,
warmupEpoch:
Int
=
0
,
maxLr:
Double
=
0.0
)
extends
Product
with
Serializable
Linear Supertypes
Serializable
,
Serializable
,
Product
,
Equals
,
AnyRef
,
Any
Ordering
Alphabetic
By inheritance
Inherited
TrainParams
Serializable
Serializable
Product
Equals
AnyRef
Any
Hide All
Show all
Learn more about member selection
Visibility
Public
All
Instance Constructors
new
TrainParams
(
folder:
String
=
"./"
,
checkpoint:
Option
[
String
] =
scala.None
,
modelSnapshot:
Option
[
String
] =
scala.None
,
stateSnapshot:
Option
[
String
] =
scala.None
,
optnet:
Boolean
=
false
,
depth:
Int
=
20
,
classes:
Int
=
10
,
shortcutType:
String
=
"A"
,
batchSize:
Int
=
128
,
nepochs:
Int
=
165
,
learningRate:
Double
=
0.1
,
weightDecay:
Double
=
1.0E-4
,
momentum:
Double
=
0.9
,
dampening:
Double
=
0.0
,
nesterov:
Boolean
=
true
,
graphModel:
Boolean
=
false
,
warmupEpoch:
Int
=
0
,
maxLr:
Double
=
0.0
)
Value Members
final
def
!=
(
arg0:
AnyRef
)
:
Boolean
Definition Classes
AnyRef
final
def
!=
(
arg0:
Any
)
:
Boolean
Definition Classes
Any
final
def
##
()
:
Int
Definition Classes
AnyRef → Any
final
def
==
(
arg0:
AnyRef
)
:
Boolean
Definition Classes
AnyRef
final
def
==
(
arg0:
Any
)
:
Boolean
Definition Classes
Any
final
def
asInstanceOf
[
T0
]
:
T0
Definition Classes
Any
val
batchSize
:
Int
val
checkpoint
:
Option
[
String
]
val
classes
:
Int
def
clone
()
:
AnyRef
Attributes
protected[
java.lang
]
Definition Classes
AnyRef
Annotations
@throws
(
...
)
val
dampening
:
Double
val
depth
:
Int
final
def
eq
(
arg0:
AnyRef
)
:
Boolean
Definition Classes
AnyRef
def
finalize
()
:
Unit
Attributes
protected[
java.lang
]
Definition Classes
AnyRef
Annotations
@throws
(
classOf[java.lang.Throwable]
)
val
folder
:
String
final
def
getClass
()
:
Class
[_]
Definition Classes
AnyRef → Any
val
graphModel
:
Boolean
final
def
isInstanceOf
[
T0
]
:
Boolean
Definition Classes
Any
val
learningRate
:
Double
val
maxLr
:
Double
val
modelSnapshot
:
Option
[
String
]
val
momentum
:
Double
final
def
ne
(
arg0:
AnyRef
)
:
Boolean
Definition Classes
AnyRef
val
nepochs
:
Int
val
nesterov
:
Boolean
final
def
notify
()
:
Unit
Definition Classes
AnyRef
final
def
notifyAll
()
:
Unit
Definition Classes
AnyRef
val
optnet
:
Boolean
val
shortcutType
:
String
val
stateSnapshot
:
Option
[
String
]
final
def
synchronized
[
T0
]
(
arg0: ⇒
T0
)
:
T0
Definition Classes
AnyRef
final
def
wait
()
:
Unit
Definition Classes
AnyRef
Annotations
@throws
(
...
)
final
def
wait
(
arg0:
Long
,
arg1:
Int
)
:
Unit
Definition Classes
AnyRef
Annotations
@throws
(
...
)
final
def
wait
(
arg0:
Long
)
:
Unit
Definition Classes
AnyRef
Annotations
@throws
(
...
)
val
warmupEpoch
:
Int
val
weightDecay
:
Double
Inherited from
Serializable
Inherited from
Serializable
Inherited from
Product
Inherited from
Equals
Inherited from
AnyRef
Inherited from
Any
Ungrouped