geomloss question about the error : arange: cannot compute length

question about the error : arange: cannot compute length

Open liangaomng opened this issue 1 year ago • 3 comments

def epsilon_schedule(p, diameter, blur, scaling): 117 r"""Creates a list of values for the temperature "epsilon" across Sinkhorn iterations. 118 119 We use an aggressive strategy with an exponential cooling (...) 140 list of float: list of values for the temperature epsilon. 141 """ 142 eps_list = ( 143 [diameter ** p] 144 + [ 145 np.exp(e) --> 146 for e in np.arange( 147 p * np.log(diameter), p * np.log(blur), p * np.log(scaling) 148 ) 149 ] 150 + [blur ** p] 151 ) 152 return eps_list

ValueError: arange: cannot compute length