Train speech language ID classification head#450

Open

am831 wants to merge 91 commits into

facebookresearch:mainfrom

am831:language_id

Commits on May 9, 2024

classification head class

am831
authored and
zrthxn
committed
finetune script progress

am831
authored and
zrthxn
committed
add comment

am831
authored and
zrthxn
committed
add layers

am831
authored and
zrthxn
committed
Model freeze und save classification head
zrthxn
committed
Implement train loop
zrthxn
committed
Implement train loop
zrthxn
committed
calc loss, class head params

am831
authored and
zrthxn
committed
Refactor
zrthxn
committed
fix errors

am831
authored and
zrthxn
committed
get vector dimensions within classification head

am831
authored and
zrthxn
committed
hidden_dim

am831
authored and
zrthxn
committed
log and capture interrupts

am831
authored and
zrthxn
committed
dataset prep and plotting loss

am831
authored and
zrthxn
committed

Commits on May 10, 2024

Commits on May 11, 2024

Model Fixes (#3)
zrthxn
authored

Commits on May 12, 2024

Merge branch 'facebookresearch:main' into language_id
am831
authored

Commits on May 13, 2024

Merge branch 'language_id' of https://github.com/am831/seamless_communication into language_id
am831
committed

Commits on May 15, 2024

classification head class

am831
authored and
zrthxn
committed
finetune script progress

am831
authored and
zrthxn
committed
add comment

am831
authored and
zrthxn
committed
add layers

am831
authored and
zrthxn
committed
Model freeze und save classification head
zrthxn
committed
Implement train loop
zrthxn
committed
Implement train loop
zrthxn
committed
calc loss, class head params

am831
authored and
zrthxn
committed
Refactor
zrthxn
committed
fix errors

am831
authored and
zrthxn
committed
get vector dimensions within classification head

am831
authored and
zrthxn
committed
hidden_dim

am831
authored and
zrthxn
committed
log and capture interrupts

am831
authored and
zrthxn
committed
dataset prep and plotting loss

am831
authored and
zrthxn
committed
Language ID Dataloader (#2)
zrthxn
committed
Model Fixes (#3)
zrthxn
committed
get embed_dim dynamically (#4)

am831
authored and
zrthxn
committed
Model Fixes
zrthxn
committed

Commits on May 16, 2024

Commits on May 17, 2024

Commits on May 18, 2024

Commits on May 19, 2024

change model to increase train loss
am831
committed

Commits on May 22, 2024

classification head class

am831
authored and
zrthxn
committed
finetune script progress

am831
authored and
zrthxn
committed
add comment

am831
authored and
zrthxn
committed
add layers

am831
authored and
zrthxn
committed
Model freeze und save classification head
zrthxn
committed
Implement train loop
zrthxn
committed
Implement train loop
zrthxn
committed
calc loss, class head params

am831
authored and
zrthxn
committed
Refactor
zrthxn
committed
fix errors

am831
authored and
zrthxn
committed
get vector dimensions within classification head

am831
authored and
zrthxn
committed
hidden_dim

am831
authored and
zrthxn
committed
log and capture interrupts

am831
authored and
zrthxn
committed
dataset prep and plotting loss

am831
authored and
zrthxn
committed
Language ID Dataloader (#2)
zrthxn
committed
Model Fixes (#3)
zrthxn
committed
classification head class

am831
authored and
zrthxn
committed
finetune script progress

am831
authored and
zrthxn
committed
add comment

am831
authored and
zrthxn
committed
add layers

am831
authored and
zrthxn
committed
Model freeze und save classification head
zrthxn
committed
Implement train loop
zrthxn
committed
Implement train loop
zrthxn
committed
calc loss, class head params

am831
authored and
zrthxn
committed
Refactor
zrthxn
committed
fix errors

am831
authored and
zrthxn
committed
get vector dimensions within classification head

am831
authored and
zrthxn
committed
hidden_dim

am831
authored and
zrthxn
committed
log and capture interrupts

am831
authored and
zrthxn
committed
dataset prep and plotting loss

am831
authored and
zrthxn
committed
Language ID Dataloader (#2)
zrthxn
committed
Model Fixes (#3)
zrthxn
committed
get embed_dim dynamically (#4)

am831
authored and
zrthxn
committed
Model Fixes
zrthxn
committed
address some feedback
zrthxn
committed
save plot as pkl

am831
authored and
zrthxn
committed
switch model to train mode

am831
authored and
zrthxn
committed
Code cleanup
zrthxn
committed
BCE loss
zrthxn
committed
Remove Label smoothing
zrthxn
committed
change model to increase train loss

am831
authored and
zrthxn
committed
lid classification head training script

mavlyutovr
authored and
zrthxn
committed
fixes

mavlyutovr
authored and
zrthxn
committed

Commits on May 25, 2024

Merge pull request #8 from zrthxn/lid/ruslan-fixes
am831
authored

Commits on Jun 3, 2024

Merge branch 'language_id' of https://github.com/am831/seamless_communication into language_id
am831
committed