braindotai

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

braindotai

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