Skip to content

Commit 7194a42

Browse files
committed
fix snapshotting with dynamic
1 parent e875fa2 commit 7194a42

File tree

10 files changed

+2
-1
lines changed

10 files changed

+2
-1
lines changed

Diff for: .gitignore

100644100755
File mode changed.

Diff for: LICENSE

100644100755
File mode changed.

Diff for: README.md

100644100755
File mode changed.

Diff for: affinity_search/NOTES

100644100755
File mode changed.

Diff for: affinity_search/makemodel.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -78,7 +78,7 @@ def add_conv_args(n,defaultwidth):
7878
add_conv_args(5,0)
7979

8080
#fully connected layer
81-
parser.add_argument('--fc_affinity_hidden',type=int,help='Hidden nodes in affinity fully connected layer; 0 for single layer',default=0,choices=(0,32,64,128,256,512,1024,2048,4096))
81+
parser.add_argument('--fc_affinity_hidden',type=int,help='Hidden nodes in affinity fully connected layer; 0 for single layer',default=0,choices=(0,16,32,64,128,256,512,1024,2048,4096))
8282
parser.add_argument('--fc_affinity_func',type=str,help="Activation function in for first affinity hidden layer",default='ReLU',choices=('ReLU','leaky','ELU','Sigmoid','TanH'))
8383
parser.add_argument('--fc_affinity_hidden2',type=int,help='Second set of hidden nodes in affinity fully connected layer; 0 for single layer',default=0,choices=(0,32,64,128,256,512,1024,2048,4096))
8484
parser.add_argument('--fc_affinity_func2',type=str,help="Activation function in for second affinity hidden layer",default='ReLU',choices=('ReLU','leaky','ELU','Sigmoid','TanH'))

Diff for: affinity_search/summaries.ipynb

100644100755
File mode changed.

Diff for: docs/README.md

100644100755
File mode changed.

Diff for: docs/Using_the_Cluster.md

100644100755
File mode changed.

Diff for: models/lenet1.template

100644100755
File mode changed.

Diff for: train.py

+1
Original file line numberDiff line numberDiff line change
@@ -603,6 +603,7 @@ def update_from_result(name, test, result):
603603
if step_reduce_cnt > args.step_end_cnt or lr < args.step_end:
604604
#end early, but run full test if needed
605605
keepsnap = True
606+
solver.snapshot()
606607
if args.reduced:
607608
training = False
608609
last_test = True

0 commit comments

Comments
 (0)