Run on Google Colab:
Connect to GPUs and Mount drive
Clone the repo (with GitHub access token: https://docs.github.com/en/authentication/keeping-your-account-and-data-secure/managing-your-personal-access-tokens#keeping-your-personal-access-tokens-secure)
Run the train command with the “—cuda” argument. Example:
! python train.py \\
--data data/en-sv/infopankki/prepared \\
--source-lang sv \\
--target-lang en \\
--save-dir assignments/01/baseline/checkpoints \\
--train-on-tiny \\
--cuda
Be careful about:
Make sure the laptop doesn’t go into sleep mode after some inactive time (because Colab might disconnect then)
There is a trick: https://stackoverflow.com/questions/57113226/how-can-i-prevent-google-colab-from-disconnecting
function ConnectButton(){
console.log("Connect pushed");
document.querySelector("#top-toolbar > colab-connectbutton").shadowRoot.querySelector("#connect").click()
}
setInterval(ConnectButton,60000);
Run on local: