Fix checkpointer bug and add checkpointer tests (#144)

[georgeyiasemis]

• Do not try to load state_dict if key is of the format <>
• Added checkpointer test