9 lines
153 B
Python
9 lines
153 B
Python
|
import torch
|
||
|
import pandas as pd
|
||
|
import torch.nn as nn
|
||
|
import torch.nn.functional as F
|
||
|
from torch.utils.data import Dataset, DataLoader
|
||
|
|
||
|
device = 'cpu'
|
||
|
|