cnn_train.py

from cnn.model import *
from cnn.music_processor import *


device = torch.device('cuda:0' if torch.cuda.is_available() else 'cpu')
net = convNet()
net = net.to(device)
        
with open('./data/pickles/train_data.pickle', mode='rb') as f:
    songs = pickle.load(f)

minibatch = 128
soundlen = 15
epoch = 100


net.train(songs=songs, minibatch=minibatch, val_song=None, epoch=epoch, device=device, soundlen=soundlen, save_place='./models/model.pth', log='./data/log/log.txt')