Many thanks ptrblck.
I’m new to coding, could you please help me with one more thing Im getting the below error and don’t know how to fix it.
RuntimeError                              Traceback (most recent call last)
 in 
----> 1 learn.fit_one_cycle(2)
~\Anaconda3\envs\fastai_v1\lib\site-packages\fastai\train.py in fit_one_cycle(learn, cyc_len, max_lr, moms, div_factor, pct_start, final_div, wd, callbacks, tot_epochs, start_epoch)
21     callbacks.append(OneCycleScheduler(learn, max_lr, moms=moms, div_factor=div_factor, pct_start=pct_start,
22                                        final_div=final_div, tot_epochs=tot_epochs, start_epoch=start_epoch))
—> 23     learn.fit(cyc_len, max_lr, wd=wd, callbacks=callbacks)
24
25 def fit_fc(learn:Learner, tot_epochs:int=1, lr:float=defaults.lr,  moms:Tuple[float,float]=(0.95,0.85), start_pct:float=0.72,
~\Anaconda3\envs\fastai_v1\lib\site-packages\fastai\basic_train.py in fit(self, epochs, lr, wd, callbacks)
198         else: self.opt.lr,self.opt.wd = lr,wd
199         callbacks = [cb(self) for cb in self.callback_fns + listify(defaults.extra_callback_fns)] + listify(callbacks)
–> 200         fit(epochs, self, metrics=self.metrics, callbacks=self.callbacks+callbacks)
201
202     def create_opt(self, lr:Floats, wd:Floats=0.)->None:
~\Anaconda3\envs\fastai_v1\lib\site-packages\fastai\basic_train.py in fit(epochs, learn, callbacks, metrics)
99             for xb,yb in progress_bar(learn.data.train_dl, parent=pbar):
100                 xb, yb = cb_handler.on_batch_begin(xb, yb)
–> 101                 loss = loss_batch(learn.model, xb, yb, learn.loss_func, learn.opt, cb_handler)
102                 if cb_handler.on_batch_end(loss): break
103
~\Anaconda3\envs\fastai_v1\lib\site-packages\fastai\basic_train.py in loss_batch(model, xb, yb, loss_func, opt, cb_handler)
24     if not is_listy(xb): xb = [xb]
25     if not is_listy(yb): yb = [yb]
—> 26     out = model(*xb)
27     out = cb_handler.on_loss_begin(out)
28
~\Anaconda3\envs\fastai_v1\lib\site-packages\torch\nn\modules\module.py in call(self, *input, **kwargs)
548             result = self._slow_forward(*input, **kwargs)
549         else:
–> 550             result = self.forward(*input, **kwargs)
551         for hook in self._forward_hooks.values():
552             hook_result = hook(self, input, result)
 in forward(self, x)
34         x = self.seq1(x)
35         x = x.view(x.size(0), -1)
—> 36         x = self.seq2(x)
37         x = self.seq3(x)
38         x = self.dropout1(x)
~\Anaconda3\envs\fastai_v1\lib\site-packages\torch\nn\modules\module.py in call(self, *input, **kwargs)
548             result = self._slow_forward(*input, **kwargs)
549         else:
–> 550             result = self.forward(*input, **kwargs)
551         for hook in self._forward_hooks.values():
552             hook_result = hook(self, input, result)
~\Anaconda3\envs\fastai_v1\lib\site-packages\torch\nn\modules\container.py in forward(self, input)
98     def forward(self, input):
99         for module in self:
–> 100             input = module(input)
101         return input
102
~\Anaconda3\envs\fastai_v1\lib\site-packages\torch\nn\modules\module.py in call(self, *input, **kwargs)
548             result = self._slow_forward(*input, **kwargs)
549         else:
–> 550             result = self.forward(*input, **kwargs)
551         for hook in self._forward_hooks.values():
552             hook_result = hook(self, input, result)
~\Anaconda3\envs\fastai_v1\lib\site-packages\torch\nn\modules\linear.py in forward(self, input)
85
86     def forward(self, input):
—> 87         return F.linear(input, self.weight, self.bias)
88
89     def extra_repr(self):
~\Anaconda3\envs\fastai_v1\lib\site-packages\torch\nn\functional.py in linear(input, weight, bias)
1608     if input.dim() == 2 and bias is not None:
1609         # fused op is marginally faster
-> 1610         ret = torch.addmm(bias, input, weight.t())
1611     else:
1612         output = input.matmul(weight.t())
RuntimeError: size mismatch, m1: [8 x 193600], m2: [254016 x 512] at C:/cb/pytorch_1000000000000/work/aten/src\THC/generic/THCTensorMathBlas.cu:283