Skip to content

Commit f6d8703

Browse files
author
Ryan Nett
committed
training time note
Signed-off-by: Ryan Nett <[email protected]>
1 parent 7bda08a commit f6d8703

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

dl4j-examples/src/main/java/org/deeplearning4j/examples/samediff/tfimport/SameDiffTransferLearningExample.java

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -244,7 +244,8 @@ public static void main(String[] args) throws Exception {
244244
// again, we reshape to the proper size as part of the data set iterator
245245
DataSetIterator trainData = new Cifar10DataSetIterator(32, new int[]{224, 224}, DataSetType.TRAIN, null, 12345);
246246

247-
//Perform fine tuning for 20 epochs. The pre-trained weights are imported as constants, and thus not trained
247+
//Perform fine tuning for 20 epochs. The pre-trained weights are imported as constants, and thus not trained.
248+
// Note that this may take a long time, especially if you try to use the CPU backend.
248249
int numEpochs = 20;
249250
History hist = sd.fit()
250251
.train(trainData, numEpochs)

0 commit comments

Comments
 (0)