your snapshot should be enough in case anything goes wrong. Ubuntu upgrade are safe anyway. In the worst case scenario, before restoring the snapshot you can also try to connect via port 1022 as suggested with

ssh -p 1022 username@your_server_ip

You can also try to keep your session alive using tmux or screen on the command to avoid accidental disconnections:

sudo apt install tmux -y && tmux new -s upgrade