Code

Training#5

Code

Draft

soumik12345 wants to merge 14 commits intoseason-1-episode-2soumik12345/jax-series:season-1-episode-2from trainingsoumik12345/jax-series:trainingCopy head branch name to clipboard

+282-12

Commits

Commits on Aug 10, 2022

• added funcion for creating train state

  soumik12345

  committed
  72e7201

  Copy full SHA for 72e7201

• updated the dataloader to get both tf dataset and iterator

  soumik12345

  committed
  02074b2

  Copy full SHA for 02074b2

• Added train step

  soumik12345

  committed
  9705b7f

  Copy full SHA for 9705b7f

• updated train_step implementation

  soumik12345

  committed
  0894913

  Copy full SHA for 0894913

• added validation step

  soumik12345

  committed
  2cd8bc7

  Copy full SHA for 2cd8bc7

Commits on Aug 11, 2022

• updated dataloader to return tf dataset, iterator and steps_per_epoch

  soumik12345

  committed
  ab651fb

  Copy full SHA for ab651fb

• started implementing train_and_validate

  soumik12345

  committed
  a8fb6bd

  Copy full SHA for a8fb6bd

Commits on Sep 1, 2022

• @SauravMaheshkar PR Review+lint

  SauravMaheshkar

  committed
  1ec0226

  Copy full SHA for 1ec0226

• fixed typo in dataloader

  soumik12345

  committed
  bf57d27

  Copy full SHA for bf57d27

• updated model initialization

  soumik12345

  committed
  5280f21

  Copy full SHA for 5280f21

• fixed typo in train_state.py

  soumik12345

  committed
  c01b016

  Copy full SHA for c01b016

• added tpu setup utility for google colab

  soumik12345

  committed
  469e1d9

  Copy full SHA for 469e1d9

• fixed imports in dataloader

  soumik12345

  committed
  375dbd3

  Copy full SHA for 375dbd3

• fixed unnecessary parameter + typo

  soumik12345

  committed
  579b169

  Copy full SHA for 579b169