NJU-RL / ACORM

25 stars 4 forks source link

24000步的时候报错 #4

Open Hkxuan opened 4 months ago

Hkxuan commented 4 months ago

ValueError: Input contains NaN, infinity or a value too large for dtype('float64').

Hkxuan commented 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]])