我正在尝试使用transfer learning
网络和PySpark
网络应用VGG-16
技术。
我只想冻结第一个m
层,然后训练其余n-m
层。我已经尝试了DeepImageFeaturizer()
库中的sparkdl
函数,但是无法选择要训练的层数:
DeepImageFeaturizer(inputCol="image", outputCol="features", modelName="VGG16")
我还看到了KerasTransformer()
函数,但不确定如何正确使用它。
我想知道:
Spark
有可能吗?KerasTransformer
函数吗?fine-tuning
来训练这种网络的所有层吗?