Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Implement a Fourier feature neural network codec #11

Merged
merged 29 commits into from
Dec 11, 2024
Merged
Changes from 1 commit
Commits
Show all changes
29 commits
Select commit Hold shift + click to select a range
74087e0
Implement a fourier feature neural network codec
juntyr Nov 28, 2024
8765ec4
Bump rand version to v0.8.5
juntyr Nov 28, 2024
e74b26f
Bump log version to v0.4.17
juntyr Nov 28, 2024
a62cf26
Bump serde version to v1.0.198
juntyr Nov 28, 2024
4e8c4fb
Work around libc 0.2.166 for now
juntyr Nov 28, 2024
4ea266e
Work around libc 0.2.166 for now v2
juntyr Nov 28, 2024
67586bf
Bump libc version to v0.2.62
juntyr Nov 28, 2024
72f5ca1
Bump libc version to v0.2.143
juntyr Nov 28, 2024
7de0aaf
Publish pre-release version
juntyr Nov 28, 2024
36afb34
Implement mini batching for the FourierNetwork codec
juntyr Dec 2, 2024
0ee32a1
Fix clippy lints
juntyr Dec 2, 2024
a31408f
Publish 0.1.0-alpha.2 pre-release version
juntyr Dec 2, 2024
dec4a57
Experiment with burn instead of candle (TODO: batching)
juntyr Dec 4, 2024
334b32c
Fix minimum versions
juntyr Dec 4, 2024
3b6e71d
Fix clippy lints
juntyr Dec 4, 2024
e26a402
Bump MSRV to 1.81, begrudgingly
juntyr Dec 4, 2024
cbc04e6
Fix clippy lints in tests
juntyr Dec 5, 2024
320324a
Further refactoring
juntyr Dec 5, 2024
544132d
Fix MSRV in numcodecs-fourier-network docs
juntyr Dec 5, 2024
aca51f2
Fix burn imports
juntyr Dec 5, 2024
54c008c
Add extra test case
juntyr Dec 5, 2024
e0a8931
Publish 0.1.0-alpha.3 pre-release version
juntyr Dec 5, 2024
a936d70
Add MAE, RMSE, and Linf metrics to epoch logs
juntyr Dec 6, 2024
4b8b386
Restore model with lowest loss after training
juntyr Dec 7, 2024
4884161
Test and fix FourierNetwork schema and config
juntyr Dec 11, 2024
103ab1e
Test RandomProjection schema and config
juntyr Dec 11, 2024
290c1a8
Fix rustfmt
juntyr Dec 11, 2024
4628f9a
Reduce the CI test matrix size
juntyr Dec 11, 2024
48262dd
Publish v0.1.0 of numcodecs-fourier-network
juntyr Dec 11, 2024
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
Fix clippy lints
juntyr committed Dec 2, 2024
commit 0ee32a1c224cb517f9d063ef4e81190ecb60eac9
1 change: 1 addition & 0 deletions codecs/fourier-network/src/lib.rs
Original file line number Diff line number Diff line change
@@ -508,6 +508,7 @@ fn fourier_mapping(xs: &Tensor, b_t: &Tensor) -> Result<Tensor, CandleError> {
Tensor::cat(&[xs_proj.sin()?, xs_proj.cos()?], D::Minus1)
}

#[allow(clippy::similar_names)] // bnX_Y and lnX_Y
fn make_model<T: FloatExt>(
num_blocks: NonZeroUsize,
fourier_features: NonZeroUsize,