fix DFLJPG,
SAE: added "rare sample booster" SAE: pixel loss replaced to smooth transition from DSSIM to PixelLoss in 15k epochs by default
This commit is contained in:
+1
-1
@@ -68,7 +68,7 @@ class devicelib:
|
||||
|
||||
@staticmethod
|
||||
def getDevicesWithAtLeastTotalMemoryGB(totalmemsize_gb):
|
||||
if not hasNVML and totalmemsize_gb <= 2:
|
||||
if not hasNVML:
|
||||
return [0]
|
||||
|
||||
result = []
|
||||
|
||||
Reference in New Issue
Block a user