10 references to FreezeModuleParams
Microsoft.ML.TorchSharp (10)
NasBert\Models\NasBertEncoder.cs (8)
131ModelUtils.FreezeModuleParams(TokenEmbedding); 132ModelUtils.FreezeModuleParams(PositionalEmbedding); 133ModelUtils.FreezeModuleParams(SegmentEmbedding); 134ModelUtils.FreezeModuleParams(EmbeddingLayerNorm); 139ModelUtils.FreezeModuleParams(Layers); 140ModelUtils.FreezeModuleParams(HiddenTransferList); 145ModelUtils.FreezeModuleParams(HiddenTransferList); 150ModelUtils.FreezeModuleParams(Layers[i]);
NasBert\Models\NasBertModel.cs (1)
44ModelUtils.FreezeModuleParams(Encoder);
Utils\ModelUtils.cs (1)
41FreezeModuleParams(module);