RemoteIndian

I am connecting to AWS instance via VS code and training a ML model. But the connection keeps dropping off after the 1st epoch. I retried several times and the execution of cells works fine until 1st epoch and then it looses connection. What could I be doing wrong?

Have you tried updating `ServerAliveInterval`  for the instance?

I am assuming you use the Microsoft's remote ssh extenstion, if yes, update your ssh config file (cmd+shift+p -&gt; `Remote.SSH: Config File`) to look something like this
```Host Titan
  HostName 192.168.1.150
  User nikhil
  ServerAliveInterval 60
  ServerAliveCountMax 30```
Update the `ip` and `user` to your instance