Commit ·
f163f2d
1
Parent(s): 5070fc9
Update resnet.py
Browse files
resnet.py
CHANGED
|
@@ -5,7 +5,7 @@ import torch.nn.functional as F
|
|
| 5 |
from io import BytesIO
|
| 6 |
import numpy as np
|
| 7 |
import os
|
| 8 |
-
from pytorch_lightning import LightningModule, Trainer
|
| 9 |
from torch import nn
|
| 10 |
from torch.nn import functional as F
|
| 11 |
from torch.utils.data import DataLoader, random_split
|
|
|
|
| 5 |
from io import BytesIO
|
| 6 |
import numpy as np
|
| 7 |
import os
|
| 8 |
+
#from pytorch_lightning import LightningModule, Trainer
|
| 9 |
from torch import nn
|
| 10 |
from torch.nn import functional as F
|
| 11 |
from torch.utils.data import DataLoader, random_split
|