11 Commits

Author	SHA1	Message	Date
Thalles Silva	5ffd2becb6	added validation loop	2020-03-10 10:35:11 -03:00
Thalles	af77a03f0b	fixed bug missing zero_grad()	2020-02-29 19:51:00 -03:00
Thalles	88dcdf6d06	support to use other pytorch datasets	2020-02-29 19:32:37 -03:00
Thalles	70bc1cbaaf	new functionality to change the similarity function	2020-02-29 15:21:04 -03:00
Thalles	67b8b5b0c1	added tensorboard support	2020-02-29 07:53:14 -03:00
Thalles	6e555f1f1c	complete the loss function description from the paper	2020-02-25 11:32:40 -03:00
Thalles	5ed8a2e453	added tensorboard support	2020-02-24 18:23:44 -03:00
Thalles Silva	289f9e54b0	improved method for computing the loss function	2020-02-20 10:02:03 -03:00
Thalles Silva	34930993d5	Update utils.py	2020-02-18 17:21:50 -03:00
Thalles	7b36fb3763	added random gaussian blur	2020-02-18 15:06:14 -03:00
Thalles	8d28ca5d31	first commit	2020-02-17 16:05:44 -03:00