Skip to content

Commit

Permalink
✨ feat(layer_seq): EmbeddingSeq (#122)
Browse files Browse the repository at this point in the history
  • Loading branch information
jean-francoisreboud authored Jun 14, 2024
1 parent ceff714 commit d97e520
Show file tree
Hide file tree
Showing 23 changed files with 2,146 additions and 145 deletions.
1 change: 1 addition & 0 deletions CHANGELOG.md
Original file line number Diff line number Diff line change
Expand Up @@ -4,6 +4,7 @@ All notable changes to this project will be documented in this file.

## [unreleased]

**layer_seq:** EmbeddingSeq ([122](https://github.com/owkin/GrAIdient/pull/122))\
🚀 **perf:** use half in Metal kernels ([121](https://github.com/owkin/GrAIdient/pull/121))\
🔨 **refactor:** handle float16 along float on GPU ([#120](https://github.com/owkin/GrAIdient/pull/120))\
🚀 **perf:** copy & generate weights faster ([119](https://github.com/owkin/GrAIdient/pull/119))\
Expand Down
2 changes: 1 addition & 1 deletion Sources/GrAIdient/Layer1D/Constant1D.swift
Original file line number Diff line number Diff line change
Expand Up @@ -21,7 +21,7 @@ public class Constant1D: Layer1D, LayerUpdate
var _wBuffers: IWeightBuffers! = nil

///
/// Buffer of gradients per sample for biases.
/// Buffer of gradients per sample.
/// Shape ~ (batch, nbNeurons).
///
var _wDeltaWeights: FloatBuffer! = nil
Expand Down
2 changes: 1 addition & 1 deletion Sources/GrAIdient/Layer2D/Constant2D.swift
Original file line number Diff line number Diff line change
Expand Up @@ -21,7 +21,7 @@ public class Constant2D: Layer2D, LayerResize, LayerUpdate
var _wBuffers: IWeightBuffers! = nil

///
/// Buffer of gradients per sample for biases.
/// Buffer of gradients per sample.
/// Shape ~ (batch, nbChannels).
///
var _wDeltaWeights: FloatBuffer! = nil
Expand Down
2 changes: 1 addition & 1 deletion Sources/GrAIdient/Layer2D/VQ2D.swift
Original file line number Diff line number Diff line change
Expand Up @@ -59,7 +59,7 @@ public class VQ2D: LayerOutput2D, LayerWeightInit
var _wBuffers: IWeightBuffers! = nil

///
/// Buffer of gradients per sample for biases.
/// Buffer of gradients per sample.
/// Shape ~ (batch, K, nbChannels).
///
var _wDeltaWeights: FloatBuffer! = nil
Expand Down
2 changes: 1 addition & 1 deletion Sources/GrAIdient/LayerSeq/ConstantSeq.swift
Original file line number Diff line number Diff line change
Expand Up @@ -505,7 +505,7 @@ public class Constant2Seq: LayerSeq, LayerUpdate
var _wBuffers: IWeightBuffers! = nil

///
/// Buffer of gradients per sample for biases.
/// Buffer of gradients per sample.
/// Shape ~ (batch, sequence, nbNeurons).
///
var _wDeltaWeights: FloatBuffer! = nil
Expand Down
Loading

0 comments on commit d97e520

Please sign in to comment.