Thank you @smth  But since simetimes I can finish one whole epoch with option shuffle=False so that I think my dataset works fine.
I’m suspecting that my dataset does not converge with block structure like resnet. As a result i’m trying the more classic VGG structure.
However, when i’m trying VGG 16 or VGG 13 structure, I’ve got the following errors:
> Traceback (most recent call last):
>   File "pytorch_finetuning.py", line 171, in <module>
>     num_ftrs = model_conv.fc8.in_features
>   File "/usr/local/lib/python3.5/dist-packages/torch/nn/modules/module.py", line 237, in __getattr__
>     return object.__getattr__(self, name)
> AttributeError: type object 'object' has no attribute '__getattr__'

I would like to ensure if VGG structure supports the call like model.fc.in_features or model.fc8.in_features