1 write to SegmentEmbedding
Microsoft.ML.TorchSharp (1)
NasBert\Models\NasBertEncoder.cs (1)
90SegmentEmbedding = numSegments > 0
6 references to SegmentEmbedding
Microsoft.ML.TorchSharp (6)
NasBert\Models\NasBertEncoder.cs (6)
100if (SegmentEmbedding != null) 102ModelUtils.InitNormal(SegmentEmbedding.weight, mean: 0.0, std: 0.02); 133ModelUtils.FreezeModuleParams(SegmentEmbedding); 210if (SegmentEmbedding != null && segmentLabels.IsNotNull()) 212var segmentEmbedding = SegmentEmbedding.forward(segmentLabels); 296SegmentEmbedding?.Dispose();