Light cnn
Compare changes
- Guillaume HEUSCH authored
[CNNTrainer] proper loading, initialization of pre-trained model, taking into account the case when the number of classes are different
+ 92
− 24
@@ -13,6 +13,7 @@ logger = bob.core.log.setup("bob.learn.pytorch")
@@ -13,6 +13,7 @@ logger = bob.core.log.setup("bob.learn.pytorch")
@@ -31,7 +32,7 @@ class CNNTrainer(object):
@@ -31,7 +32,7 @@ class CNNTrainer(object):
@@ -44,9 +45,12 @@ class CNNTrainer(object):
@@ -44,9 +45,12 @@ class CNNTrainer(object):
@@ -56,31 +60,88 @@ class CNNTrainer(object):
@@ -56,31 +60,88 @@ class CNNTrainer(object):
@@ -133,8 +194,15 @@ class CNNTrainer(object):
@@ -133,8 +194,15 @@ class CNNTrainer(object):
logger.info('Starting training at epoch {}, iteration {} - last loss value is {}'.format(start_epoch, start_iter, losses[-1]))