Skip to content

Update trainSpeakerNet.py#191

Open
mst-rajatmishra wants to merge 1 commit into
clovaai:masterfrom
mst-rajatmishra:master
Open

Update trainSpeakerNet.py#191
mst-rajatmishra wants to merge 1 commit into
clovaai:masterfrom
mst-rajatmishra:master

Conversation

@mst-rajatmishra

Copy link
Copy Markdown

hi,
I Tried making some changes,

  1. Introduced logging to replace print statements. This allows for more granular control of the log levels (e.g., INFO, DEBUG, ERROR).
  2. Added autocast and GradScaler for mixed precision training in the trainer loop.
  3. Used pin_memory=True for DataLoader to speed up data transfer to the GPU.
  4. Updated the learning rate decay process to check for test_interval.
  5. Improved the comments and clarified some sections of the code.

Thank You,
RAJAT MISHRA

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant