Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
41 commits
Select commit Hold shift + click to select a range
1b133fa
Add tests
JakobHavtorn Jun 9, 2023
b30cd57
Remove prints
JakobHavtorn Jun 9, 2023
894c8f7
Conv module update and scraping
borgholt Jun 19, 2023
8bbd8da
Scraping and assignment via testing done.
borgholt Jun 19, 2023
bc7ad50
Merge branch 'main' into lazy_metadata
JakobHavtorn Jun 22, 2023
011fd32
Merge remote-tracking branch 'origin/conv-module' into wav2letter
JakobHavtorn Jun 22, 2023
6ca11af
Add outer level patch method
JakobHavtorn Jun 22, 2023
59ddca7
Add test for conv patch
JakobHavtorn Jun 22, 2023
f9b03a2
Fix linting
JakobHavtorn Jun 22, 2023
d82e041
Add license
JakobHavtorn Jun 22, 2023
a40c7bc
Format pointwise_ops
JakobHavtorn Jun 22, 2023
c022efe
Start implementing RNN
JakobHavtorn Jun 22, 2023
03bb2d0
Remove wav2letter test
JakobHavtorn Jun 23, 2023
a4c5711
Update tests and fix deepcopy
JakobHavtorn Jun 23, 2023
0d8e4f8
Reformat listloaders.py
JakobHavtorn Jun 23, 2023
83aed63
Fix linting
JakobHavtorn Jun 23, 2023
9e14694
Merge branch 'wav2letter' into rnn and implement StateStore
JakobHavtorn Jun 23, 2023
9a4f70f
Add chunk_indices to equal and deepcopy
JakobHavtorn Jun 27, 2023
26a150e
Add chunk_indices to indexing
JakobHavtorn Jun 27, 2023
2a0103c
Format with black
JakobHavtorn Jun 27, 2023
e8bc8c1
Add questions regarding extra tokens for Transformers
JakobHavtorn Jun 27, 2023
c9d276e
Merge branches 'lazy_metadata' and 'rnn' of github.com:borgholt/dream…
JakobHavtorn Jun 27, 2023
5719824
Updates for testing
JakobHavtorn Jun 27, 2023
549739b
Fix typing state ids as tuple
JakobHavtorn Jun 27, 2023
5e19c57
Remove LazyInit tests
JakobHavtorn Jun 27, 2023
aaffc2b
Add test of time saving from lazy initialisation
JakobHavtorn Jun 27, 2023
ccb6c04
Cleanup
JakobHavtorn Jun 27, 2023
890955e
Rename test class to avoid pytest warning
JakobHavtorn Jun 27, 2023
8194128
Fix warnings
JakobHavtorn Jun 27, 2023
e4ea535
Merge branch 'lazy_metadata' into rnn
JakobHavtorn Jun 28, 2023
74249c2
Add missing lazy classes back into tensor.py
JakobHavtorn Jun 28, 2023
3c89749
Add test command
JakobHavtorn Jul 4, 2023
ef5d0fb
Add onnx
JakobHavtorn Jul 4, 2023
6e6e015
Fix formatting
JakobHavtorn Jul 4, 2023
e3cd110
Add RNN methods
JakobHavtorn Jul 4, 2023
f4b6d87
Fix chunk indices
JakobHavtorn Jul 4, 2023
f84199d
Add append to list option
JakobHavtorn Jul 4, 2023
d35f8da
Add a few type hints
JakobHavtorn Jul 4, 2023
52dae47
Add partially working RNN
JakobHavtorn Jul 4, 2023
a85e7d9
Add test for unbind
JakobHavtorn Jul 4, 2023
a1f878c
Add conv test case for structured batches
JakobHavtorn Jul 4, 2023
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
19 changes: 19 additions & 0 deletions LICENSE
Original file line number Diff line number Diff line change
@@ -0,0 +1,19 @@
Copyright (c) 2023 Corti.ai

Permission is hereby granted, free of charge, to any person obtaining a copy
of this software and associated documentation files (the "Software"), to deal
in the Software without restriction, including without limitation the rights
to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
copies of the Software, and to permit persons to whom the Software is
furnished to do so, subject to the following conditions:

The above copyright notice and this permission notice shall be included in all
copies or substantial portions of the Software.

THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
SOFTWARE.
11 changes: 10 additions & 1 deletion README.md
Original file line number Diff line number Diff line change
Expand Up @@ -153,8 +153,12 @@ PyTorch models are typically trained and evaluated on batches of data. However,
- Combining batch or length dimensions with one or more other dimensions into a single dimension using e.g. `torch.reshape`, `torch.flatten` or masked indexing.
- Options:
- Fail outright.
- Fallback to a regular `torch.Tensor`.
- Fallback to a regular `torch.Tensor`. <-- Chose this one.
- Fallback to a different tensor subclass that is identical in behaviour to `torch.Tensor` but carries the frozen `StreamMetadata` along.
- Support loading/saving of named tensors by custom `__reduce__` or `__reduce_ex__`.
- How do we deal with
- Special tokens concatenated to the input? E.g. "translate" and "language" tokens in Whisper?
- Learnable tokens concatenated to the input sequence before an MHSA layer?

## Can we use DreamStream for training?

Expand Down Expand Up @@ -185,6 +189,11 @@ pip install -r requirements.txt
```


## Run tests
```bash
pytest -sv --cov=dreamstream --cov-report=term -p no:pytest_wampy tests/test_tensor.py::TestUnbind
```


```python

Expand Down
166 changes: 166 additions & 0 deletions doc_scrape/lists/default-valid-pointwise-ops-2023_06_19-16_13_42.txt
Original file line number Diff line number Diff line change
@@ -0,0 +1,166 @@
abs
Tensor.abs
absolute
Tensor.absolute
acos
Tensor.acos
arccos
Tensor.arccos
acosh
Tensor.acosh
arccosh
Tensor.arccosh
add
Tensor.add
addcdiv
Tensor.addcdiv
addcmul
Tensor.addcmul
angle
Tensor.angle
asin
Tensor.asin
arcsin
Tensor.arcsin
asinh
Tensor.asinh
arcsinh
Tensor.arcsinh
atan
Tensor.atan
arctan
Tensor.arctan
atanh
Tensor.atanh
arctanh
Tensor.arctanh
atan2
Tensor.atan2
arctan2
Tensor.arctan2
bitwise_not
Tensor.bitwise_not
ceil
Tensor.ceil
clamp
Tensor.clamp
clip
Tensor.clip
conj_physical
Tensor.conj_physical
cos
Tensor.cos
cosh
Tensor.cosh
deg2rad
Tensor.deg2rad
div
Tensor.div
divide
Tensor.divide
digamma
Tensor.digamma
erf
Tensor.erf
erfc
Tensor.erfc
erfinv
Tensor.erfinv
exp
Tensor.exp
expm1
Tensor.expm1
fix
Tensor.fix
float_power
Tensor.float_power
floor
Tensor.floor
floor_divide
Tensor.floor_divide
frac
Tensor.frac
ldexp
Tensor.ldexp
lgamma
Tensor.lgamma
log
Tensor.log
log10
Tensor.log10
log1p
Tensor.log1p
log2
Tensor.log2
logaddexp
Tensor.logaddexp
logaddexp2
Tensor.logaddexp2
logical_and
Tensor.logical_and
logical_not
Tensor.logical_not
logical_or
Tensor.logical_or
logical_xor
Tensor.logical_xor
hypot
Tensor.hypot
i0
Tensor.i0
igamma
Tensor.igamma
igammac
Tensor.igammac
mul
Tensor.mul
multiply
Tensor.multiply
neg
Tensor.neg
negative
Tensor.negative
nextafter
Tensor.nextafter
polygamma
Tensor.polygamma
positive
Tensor.positive
pow
Tensor.pow
rad2deg
Tensor.rad2deg
reciprocal
Tensor.reciprocal
round
Tensor.round
rsqrt
Tensor.rsqrt
sigmoid
Tensor.sigmoid
sign
Tensor.sign
signbit
Tensor.signbit
sin
Tensor.sin
sinh
Tensor.sinh
softmax
Tensor.softmax
sqrt
Tensor.sqrt
square
Tensor.square
sub
Tensor.sub
subtract
Tensor.subtract
tan
Tensor.tan
tanh
Tensor.tanh
true_divide
Tensor.true_divide
trunc
Tensor.trunc
Original file line number Diff line number Diff line change
@@ -0,0 +1,23 @@
Tensor.bitwise_and_
Tensor.bitwise_or_
Tensor.bitwise_xor_
Tensor.bitwise_left_shift_
Tensor.bitwise_right_shift_
conj_physical_
Tensor.conj_physical_
Tensor.copysign_
exp2_
Tensor.exp2_
Tensor.fmod_
Tensor.lerp_
logit_
Tensor.logit_
Tensor.mvlgamma_
nan_to_num_
Tensor.nan_to_num_
Tensor.remainder_
Tensor.sgn_
sinc_
Tensor.sinc_
xlogy_
Tensor.xlogy_
32 changes: 32 additions & 0 deletions doc_scrape/lists/recouple-pointwise-ops-2023_06_19-16_13_42.txt
Original file line number Diff line number Diff line change
@@ -0,0 +1,32 @@
bitwise_and
Tensor.bitwise_and
bitwise_or
Tensor.bitwise_or
bitwise_xor
Tensor.bitwise_xor
bitwise_left_shift
Tensor.bitwise_left_shift
bitwise_right_shift
Tensor.bitwise_right_shift
copysign
Tensor.copysign
exp2
Tensor.exp2
fmod
Tensor.fmod
lerp
Tensor.lerp
logit
Tensor.logit
mvlgamma
Tensor.mvlgamma
nan_to_num
Tensor.nan_to_num
remainder
Tensor.remainder
sgn
Tensor.sgn
sinc
Tensor.sinc
xlogy
Tensor.xlogy
122 changes: 122 additions & 0 deletions doc_scrape/lists/valid-pointwise-ops-2023_06_19-16_13_42.txt
Original file line number Diff line number Diff line change
@@ -0,0 +1,122 @@
abs_
Tensor.abs_
Tensor.absolute_
acos_
Tensor.acos_
arccos_
Tensor.arccos_
acosh_
Tensor.acosh_
arccosh_
Tensor.arccosh_
Tensor.add_
Tensor.addcdiv_
Tensor.addcmul_
asin_
Tensor.asin_
arcsin_
Tensor.arcsin_
asinh_
Tensor.asinh_
arcsinh_
Tensor.arcsinh_
atan_
Tensor.atan_
arctan_
Tensor.arctan_
atanh_
Tensor.atanh_
arctanh_
Tensor.arctanh_
Tensor.atan2_
Tensor.arctan2_
Tensor.bitwise_not_
ceil_
Tensor.ceil_
clamp_
Tensor.clamp_
clip_
Tensor.clip_
cos_
Tensor.cos_
cosh_
Tensor.cosh_
deg2rad_
Tensor.deg2rad_
Tensor.div_
Tensor.divide_
Tensor.digamma_
erf_
Tensor.erf_
erfc_
Tensor.erfc_
Tensor.erfinv_
exp_
Tensor.exp_
expm1_
Tensor.expm1_
fix_
Tensor.fix_
Tensor.float_power_
floor_
Tensor.floor_
Tensor.floor_divide_
frac_
Tensor.frac_
ldexp_
Tensor.ldexp_
Tensor.lgamma_
log_
Tensor.log_
log10_
Tensor.log10_
log1p_
Tensor.log1p_
log2_
Tensor.log2_
Tensor.logical_and_
Tensor.logical_not_
Tensor.logical_or_
Tensor.logical_xor_
Tensor.hypot_
i0_
Tensor.i0_
Tensor.igamma_
Tensor.igammac_
Tensor.mul_
Tensor.multiply_
neg_
Tensor.neg_
negative_
Tensor.negative_
Tensor.nextafter_
Tensor.polygamma_
Tensor.pow_
rad2deg_
Tensor.rad2deg_
reciprocal_
Tensor.reciprocal_
round_
Tensor.round_
rsqrt_
Tensor.rsqrt_
sigmoid_
Tensor.sigmoid_
Tensor.sign_
sin_
Tensor.sin_
sinh_
Tensor.sinh_
sqrt_
Tensor.sqrt_
square_
Tensor.square_
Tensor.sub_
Tensor.subtract_
tan_
Tensor.tan_
tanh_
Tensor.tanh_
Tensor.true_divide_
trunc_
Tensor.trunc_
Loading