Skip to content

Instantly share code, notes, and snippets.

Avatar
👨‍💻
Cooking stuff

Imad E. imadelh

👨‍💻
Cooking stuff
View GitHub Profile
@imadelh
imadelh / cnn.py
Created Jul 28, 2019
Training CNN
View cnn.py
import torch
from torchvision import datasets, transforms
import matplotlib.pyplot as plt
import torch.nn as nn
import torch.nn.functional as F
import torch.optim as optim
import logging
import numpy as np
# Define transformers
@imadelh
imadelh / mydocker.sh
Last active Jun 19, 2022
Nvidia Docker - Jetson Nano
View mydocker.sh
#!/bin/bash
# Copied and modified from https://github.com/Technica-Corporation/Tegra-Docker
# Copyright (c) 2017, Technica Corporation. All rights reserved.
NV_LIBS="/usr/lib/aarch64-linux-gnu \
/usr/local/cuda/lib64 \
/usr/local/cuda \
/usr/src/tensorrt \
/usr/local/cuda-10.0 \
/usr/include \