Skip to content

Instantly share code, notes, and snippets.

View braindotai's full-sized avatar
🏠
Working from home

Rishik Mourya braindotai

🏠
Working from home
View GitHub Profile
import torch
from torch import nn, optim
import torch.nn.functional as F
from torch.utils.data import DataLoader
from torchvision.datasets import ImageFolder
from torchvision import transforms
from torchsummary import summary
import wandb
PS1='\[\e[107;30m\]' # change to bold white
PS1+=' \u' # user
# PS1+='\[\e[0;92m\]' # change to bold green
# PS1+='@\h' # device
PS1+='\[\e[1;37m\]' # change to white
PS1+=' ' # user
PS1+='\[\e[97;104m\]' # change to bold blue
# PS1='\[\e[1;34m\]' # change to bold blue
PS1+=' ~/\W ' # current working directory
PS1+='\[\e[0;37m\]' # change color