Open Hkxuan opened 4 months ago
运行clusters_labels = KMeans(n_clusters=self.cluster_num).fit(agentembedding[idx]).labels 这行代码到200000多步时agent_embedding[idx]会变成tensor([[nan, nan, nan, ..., nan, nan, nan], [nan, nan, nan, ..., nan, nan, nan], [nan, nan, nan, ..., nan, nan, nan], ..., [nan, nan, nan, ..., nan, nan, nan], [nan, nan, nan, ..., nan, nan, nan], [nan, nan, nan, ..., nan, nan, nan]])
ValueError: Input contains NaN, infinity or a value too large for dtype('float64').