## Introduction
This is an individual module, which is mainly for **pytorch CNN** training.
Moreover, it also supports some awesome features: saving model, saving training process, plotting figures and so on...
## Install
`pip install Fau-tools`
## Usage
### import
The following code is recommended.
```python
import Fau_tools
from Fau_tools import torch_tools
```
### quick start
The tutor will use a simple example to help you get started quickly!
**The following example uses Fau-tools to train a model in MNIST hand-written digits dataset.**
```python
import torch
import torch.utils.data as tdata
import torchvision
from torch import nn
import Fau_tools
from Fau_tools import torch_tools
# A simple CNN network
class CNN(nn.Module):
def __init__(self):
super().__init__()
self.conv = nn.Sequential(
nn.Conv2d(1, 16, 3, 1, 1), # -> (16, 28, 28)
nn.ReLU(),
nn.MaxPool2d(2), # -> (16, 14, 14)
nn.Conv2d(16, 32, 3, 1, 1), # -> (32, 14, 14)
nn.ReLU(),
nn.MaxPool2d(2) # -> (32, 7, 7)
)
self.output = nn.Linear(32 * 7 * 7, 10)
def forward(self, x):
x = self.conv(x)
x = x.flatten(1) # same as x = x.view(x.size(0), -1)
return self.output(x)
# Hyper Parameters definition
EPOCH = 10
LR = 1E-3
BATCH_SIZE = 1024
# Load dataset
TRAIN_DATA = torchvision.datasets.MNIST('Datasets', True, torchvision.transforms.ToTensor(), download=True)
TEST_DATA = torchvision.datasets.MNIST('Datasets', False, torchvision.transforms.ToTensor())
# Get data loader
train_loader = tdata.DataLoader(TRAIN_DATA, BATCH_SIZE, True)
test_loader = tdata.DataLoader(TEST_DATA, BATCH_SIZE)
# Initialize model, optimizer and loss function
model = CNN()
optimizer = torch.optim.Adam(model.parameters(), LR)
loss_function = nn.CrossEntropyLoss()
# Train!
torch_tools.torch_train(model, train_loader, test_loader, optimizer, loss_function, EPOCH, name="MNIST")
# the last parameter is the name for saving model and training process.
```
Now, we can run the python file, and the training process will be visualized, just like the following picture.
> Three files named `MNIST_9846.pth`, `MNIST_9846.csv` and `MNIST_9846.txt` will be saved.
>
> The first file is the trained model.
>
> The second file records the training process, which you can use matplotlib to visualize it.
>
> The third file saves some hyper parameters about the training.
---
The above is the primary usage of this tool, but there are also some other snazzy features, which will be introduced later.
## END
Hope you could like it! And welcome issues and pull requests.
