from model import *

fc_model.train(model, trainloader, testloader, criterion, optimizer, epochs=2)

print(f'Our mode: \n\n{model}\n')
print(f'The state dict keys: \n\n{model.state_dict().keys()}')

torch.save(model.state_dict(), 'our_model.pth')

570360eb771eea6bd836fe7d07eb520041062d47
<mask token>

<mask token>
fc_model.train(model, trainloader, testloader, criterion, optimizer, epochs=2)
print(f"""Our mode: 

{model}
""")
print(f"""The state dict keys: 

{model.state_dict().keys()}""")
torch.save(model.state_dict(), 'our_model.pth')

from model import *
fc_model.train(model, trainloader, testloader, criterion, optimizer, epochs=2)
print(f"""Our mode: 

{model}
""")
print(f"""The state dict keys: 

{model.state_dict().keys()}""")
torch.save(model.state_dict(), 'our_model.pth')

from model import *

fc_model.train(model, trainloader, testloader, criterion, optimizer, epochs=2)

print(f'Our mode: \n\n{model}\n')
print(f'The state dict keys: \n\n{model.state_dict().keys()}')

torch.save(model.state_dict(), 'our_model.pth')
