def __call__()

in touch_charts/recon.py [0:0]

• 18 lines of code
• 1 McCabe index (conditional complexity)

	def __call__(self) -> float:
		self.encoder = models.Encoder(self.args)
		self.encoder.cuda()
		params = list(self.encoder.parameters())
		self.optimizer = optim.Adam(params, lr=self.args.lr, weight_decay=0)
		writer = SummaryWriter(os.path.join('experiments/tensorboard/', args.exp_type ))

		train_loader, valid_loaders = self.get_loaders()

		if self.args.eval:
			self.load('')
			with torch.no_grad():
				self.validate(valid_loaders, writer)
			exit()
		for epoch in range(self.args.epochs):
			self.epoch = epoch
			self.train(train_loader, writer)
			with torch.no_grad():
				self.validate(valid_loaders, writer)
			self.check_values()