Skip to content

Instantly share code, notes, and snippets.

View Wsine's full-sized avatar
:octocat:
Feel free to contact me

Jankin Wei Wsine

:octocat:
Feel free to contact me
View GitHub Profile
import unittest
import torch
import torchvision.models as models
class TestPytorchLayers(unittest.TestCase):
def setUp(self):
if hasattr(torch.backends, 'mps') and torch.backends.mps.is_available(): # type: ignore
self.device = torch.device('mps')
#!/bin/bash
# Script for installing tmux on systems where you don't have root access.
# tmux will be installed in $HOME/bin.
# It's assumed that wget and a C/C++ compiler are installed.
# exit on error
set -e
TMUX_VERSION=2.1