# !conda install numpy pandas pytorch torchvision cpuonly -c pytorch -y
# !pip install matplotlib --upgrade --quiet
!pip install jovian --upgrade --quiet

import jovian
import torch
import torchvision
import numpy as np
import matplotlib.pyplot as plt
import torch.nn as nn
import torch.nn.functional as F
from torchvision.datasets import CIFAR10
from torchvision.transforms import ToTensor
from torchvision.utils import make_grid
from torch.utils.data.dataloader import DataLoader
from torch.utils.data import random_split
%matplotlib inline

# Project name used for jovian.commit
project_name = 'identify-the-dance-form'

weights_fname = 'Churn_Modelling.csv'
sub_fname = 'Musical_instruments_reviews.csv'

jovian.log_hyperparams(arch='resnet152', 
                       epochs=2*10, 
                       lr=0.001, 
                       scheduler='one-cycle', 
                       weight_decay=1e-3, 
                       grad_clip=0.1,
                       opt='sdg')