Update src/loss.py

src/loss.py

@@ -98,7 +98,7 @@ class LossScheduler:
 		A.e_prev.append(model_output)
 		if timestep==A.timesteps[-1]:
 			with torch.enable_grad():
-				orig_output = torch.load(f"/home/mbhat/edge-maxxing/miner/miner/latents/latent_oig_{image_count}_20.pth")
+				orig_output = torch.load(f"/home/mbhat/edge-maxxing/miner/miner/latents/latent_orig_{image_count}_20.pth")
 				A.optimizer.zero_grad()
 				C=A.model(B,A.xT,A.e_prev)
 				L=torch.nn.functional.mse_loss(C, orig_output)