Update CFD_DA_seg_trainer.py

FupingWu90 · Dec 9, 2020 · 04f0919 · 04f0919
1 parent fb90646
commit 04f0919
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/CFD_DA_seg_trainer.py b/CFD_DA_seg_trainer.py
@@ -22,7 +22,7 @@
 matplotlib.use('Agg')
 import matplotlib.pyplot as plt
 import random
-
+import argparse
 
 
 def ADA_Train( Train_LoaderA,Train_LoaderB,encoder,decoderB,decoderBdown2,decoderBdown4,gate,DistanceNet,lr,kldlamda,predlamda,bcelamda,dislamda,epoch,optim, savedir):